Struct GroupNormFuncOptions

• Defined in File normalization.h

Struct Documentation

struct GroupNormFuncOptions

Options for torch::nn::functional::group_norm.

Example:

namespace F = torch::nn::functional;
F::group_norm(input, F::GroupNormFuncOptions(2).eps(2e-5));

Public Functions

GroupNormFuncOptions(int64_t num_groups)

inline auto num_groups(const int64_t &new_num_groups) -> decltype(*this)

number of groups to separate the channels into

inline auto num_groups(int64_t &&new_num_groups) -> decltype(*this)

inline const int64_t &num_groups() const noexcept

inline int64_t &num_groups() noexcept

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

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

inline const Tensor &weight() const noexcept

inline Tensor &weight() noexcept

inline auto bias(const Tensor &new_bias) -> decltype(*this)

inline auto bias(Tensor &&new_bias) -> decltype(*this)

inline const Tensor &bias() const noexcept

inline Tensor &bias() noexcept

inline auto eps(const double &new_eps) -> decltype(*this)

a value added to the denominator for numerical stability. Default: 1e-5

inline auto eps(double &&new_eps) -> decltype(*this)

inline const double &eps() const noexcept

inline double &eps() noexcept