Struct BCELossOptions

• Defined in File loss.h

Struct Documentation

struct BCELossOptions

Options for the BCELoss module.

Example:

BCELoss model(BCELossOptions().reduction(torch::kNone).weight(weight));

Public Types

typedef std::variant<enumtype::kNone, enumtype::kMean, enumtype::kSum> reduction_t

Public Functions

inline auto weight(const Tensor &new_weight) -> decltype(*this)

A manual rescaling weight given to the loss of each batch element.

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'. Default: 'mean'

inline auto reduction(reduction_t &&new_reduction) -> decltype(*this)

inline const reduction_t &reduction() const noexcept

inline reduction_t &reduction() noexcept