Struct MarginRankingLossOptions#
Defined in File loss.h
Page Contents
Struct Documentation#
-
struct MarginRankingLossOptions#
Options for the
MarginRankingLoss
module.Example:
MarginRankingLoss model(MarginRankingLossOptions().margin(0.5).reduction(torch::kSum));
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#
-
inline auto margin(const double &new_margin) -> decltype(*this)#