Struct MultiLabelSoftMarginLossOptions¶
Defined in File loss.h
Page Contents
Struct Documentation¶
-
struct MultiLabelSoftMarginLossOptions¶
Options for the
MultiLabelSoftMarginLoss
module.Example:
MultiLabelSoftMarginLoss model(MultiLabelSoftMarginLossOptions().reduction(torch::kNone).weight(weight));
Public Functions
-
inline auto 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.
-
inline auto weight(Tensor &&new_weight) -> decltype(*this)¶
-
inline const Tensor &weight() const noexcept¶
-
inline Tensor &weight() noexcept¶
-
inline auto 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’
-
inline auto reduction(reduction_t &&new_reduction) -> decltype(*this)¶
-
inline const reduction_t &reduction() const noexcept¶
-
inline reduction_t &reduction() noexcept¶
-
inline auto weight(const Tensor &new_weight) -> decltype(*this)¶