Struct PoissonNLLLossOptions#
Defined in File loss.h
Page Contents
Struct Documentation#
-
struct PoissonNLLLossOptions#
Options for the
PoissonNLLLoss
module.Example:
PoissonNLLLoss model(PoissonNLLLossOptions().log_input(false).full(true).eps(0.42).reduction(torch::kSum));
Public Functions
-
inline auto log_input(const bool &new_log_input) -> decltype(*this)#
if true the loss is computed as
exp(input) - target * input
, if false the loss isinput - target * log(input + eps)
.
-
inline auto log_input(bool &&new_log_input) -> decltype(*this)#
-
inline const bool &log_input() const noexcept#
-
inline bool &log_input() noexcept#
-
inline auto full(const bool &new_full) -> decltype(*this)#
whether to compute full loss, i.e.
to add the Stirling approximation term target * log(target) - target + 0.5 * log(2 * pi * target).
-
inline auto full(bool &&new_full) -> decltype(*this)#
-
inline const bool &full() const noexcept#
-
inline bool &full() noexcept#
-
inline auto eps(const double &new_eps) -> decltype(*this)#
Small value to avoid evaluation of
log(0)
whenlog_input = false
.Default: 1e-8
-
inline auto eps(double &&new_eps) -> decltype(*this)#
-
inline const double &eps() const noexcept#
-
inline double &eps() 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 log_input(const bool &new_log_input) -> decltype(*this)#