Rate this Page

Struct MarginRankingLossOptions#

Page Contents

Struct Documentation#

struct MarginRankingLossOptions#

Options for the MarginRankingLoss module.

Example:

MarginRankingLoss
model(MarginRankingLossOptions().margin(0.5).reduction(torch::kSum));

Public Types

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

Public Functions

inline auto margin(const double &new_margin) -> decltype(*this)#

Has a default value of 0.

inline auto margin(double &&new_margin) -> decltype(*this)#
inline const double &margin() const noexcept#
inline double &margin() noexcept#
inline auto reduction(const reduction_t &new_reduction) -> decltype(*this)#

Specifies the reduction to apply to the output. Default: Mean.

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