ClikaRT::nn::BCEWithLogitsLossOptions
struct
Header: ClikaRT/nn/generated/loss.h
Settings of BCEWithLogitsLoss: chain the setters, or assign the fields.
Member functions
weight(OptionalTensor)
BCEWithLogitsLossOptions& weight(OptionalTensor v)
Sets weight, a per-element rescaling weight, broadcast against input; unset weighs 1.
Declared in ClikaRT/nn/generated/loss.h, line 500
weight()
const OptionalTensor& weight() const
Reads weight.
Declared in ClikaRT/nn/generated/loss.h, line 505
reduction(ops::Reduction)
BCEWithLogitsLossOptions& reduction(ops::Reduction v)
Sets reduction, how the elementwise losses reduce.
Declared in ClikaRT/nn/generated/loss.h, line 511
reduction()
ops::Reduction reduction() const
Reads reduction.
Declared in ClikaRT/nn/generated/loss.h, line 513
pos_weight(OptionalTensor)
BCEWithLogitsLossOptions& pos_weight(OptionalTensor v)
Sets pos_weight, a weight on the positive class, broadcast against input; unset weighs 1.
Declared in ClikaRT/nn/generated/loss.h, line 519
pos_weight()
const OptionalTensor& pos_weight() const
Reads pos_weight.
Declared in ClikaRT/nn/generated/loss.h, line 524
Data members
weight_
OptionalTensor weight_ = {}
A per-element rescaling weight, broadcast against input; unset weighs 1. Default {}.
Declared in ClikaRT/nn/generated/loss.h, line 508
reduction_
ops::Reduction reduction_ = ops::Reduction::Mean
How the elementwise losses reduce. Default ops::Reduction::Mean.
Declared in ClikaRT/nn/generated/loss.h, line 515
pos_weight_
OptionalTensor pos_weight_ = {}
A weight on the positive class, broadcast against input; unset weighs 1. Default {}.
Declared in ClikaRT/nn/generated/loss.h, line 527