Defined in File loss.h
Options for the
weight(const Tensor &new_weight) -> decltype(*this)¶
A manual rescaling weight given to each class.
If given, it has to be a Tensor of size
C. Otherwise, it is treated as if having all ones.
weight(Tensor &&new_weight) -> decltype(*this)¶
const Tensor &
weight() const noexcept¶
reduction(const reduction_t &new_reduction) -> decltype(*this)¶
Specifies the reduction to apply to the output: ‘none’ | ‘mean’ | ‘sum’.
‘none’: no reduction will be applied, ‘mean’: the sum of the output will be divided by the number of elements in the output, ‘sum’: the output will be summed. Default: ‘mean’