@Namespace(value="torch::nn") @NoOffset @Properties(inherit=torch.class) public class BCEWithLogitsLossImpl extends BCEWithLogitsLossImplCloneable
Sigmoid
layer and the BCELoss
in one single
class. This version is more numerically stable than using a plain Sigmoid
followed by a BCELoss
as, by combining the operations into one layer,
we take advantage of the log-sum-exp trick for numerical stability.
See https://pytorch.org/docs/master/nn.html#torch.nn.BCEWithLogitsLoss to
learn about the exact behavior of this module.
See the documentation for torch::nn::BCEWithLogitsLossOptions
class to
learn what constructor arguments are supported for this module.
Example:
BCEWithLogitsLoss
model(BCEWithLogitsLossOptions().reduction(torch::kNone).weight(weight));
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
BCEWithLogitsLossImpl() |
BCEWithLogitsLossImpl(BCEWithLogitsLossOptions options_) |
BCEWithLogitsLossImpl(Module pointer)
Downcast constructor.
|
BCEWithLogitsLossImpl(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
Tensor |
forward(Tensor input,
Tensor target) |
BCEWithLogitsLossOptions |
options()
The options with which this
Module was constructed. |
BCEWithLogitsLossImpl |
options(BCEWithLogitsLossOptions setter) |
Tensor |
pos_weight()
A weight of positive examples.
|
BCEWithLogitsLossImpl |
pos_weight(Tensor setter) |
void |
pretty_print(Pointer stream)
Pretty prints the
BCEWithLogitsLoss module into the given stream . |
void |
reset()
reset() must perform initialization of all members with reference
semantics, most importantly parameters, buffers and submodules. |
Tensor |
weight()
A manual rescaling weight given to the loss of each batch element.
|
BCEWithLogitsLossImpl |
weight(Tensor setter) |
asModule, asModule, clone, clone
apply, apply, apply, apply, apply, apply, apply, apply, buffers, buffers, children, eval, is_serializable, is_training, load, modules, modules, name, named_buffers, named_buffers, named_children, named_modules, named_modules, named_modules, named_parameters, named_parameters, parameters, parameters, register_buffer, register_buffer, register_module, register_module, register_parameter, register_parameter, register_parameter, register_parameter, save, shiftLeft, to, to, to, train, unregister_module, unregister_module, zero_grad
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public BCEWithLogitsLossImpl(Pointer p)
Pointer(Pointer)
.public BCEWithLogitsLossImpl(Module pointer)
public BCEWithLogitsLossImpl(@ByVal(nullValue="torch::nn::BCEWithLogitsLossOptions{}") BCEWithLogitsLossOptions options_)
public BCEWithLogitsLossImpl()
public void reset()
BCEWithLogitsLossImplCloneable
reset()
must perform initialization of all members with reference
semantics, most importantly parameters, buffers and submodules.reset
in class BCEWithLogitsLossImplCloneable
public void pretty_print(@Cast(value="std::ostream*") @ByRef Pointer stream)
BCEWithLogitsLoss
module into the given stream
.pretty_print
in class Module
@ByRef public BCEWithLogitsLossOptions options()
Module
was constructed.public BCEWithLogitsLossImpl options(BCEWithLogitsLossOptions setter)
@ByRef public Tensor weight()
public BCEWithLogitsLossImpl weight(Tensor setter)
public BCEWithLogitsLossImpl pos_weight(Tensor setter)
Copyright © 2024. All rights reserved.