Struct SoftminFuncOptions¶
Defined in File activation.h
Page Contents
Struct Documentation¶
-
struct SoftminFuncOptions¶
Options for
torch::nn::functional::softmin
.Example:
namespace F = torch::nn::functional; F::softmin(input, F::SoftminFuncOptions(1));
Public Functions
-
SoftminFuncOptions(int64_t dim)¶
-
inline auto dim(const int64_t &new_dim) -> decltype(*this)¶
Dimension along which Softmin will be computed.
-
inline auto dim(int64_t &&new_dim) -> decltype(*this)¶
-
inline const int64_t &dim() const noexcept¶
-
inline int64_t &dim() noexcept¶
-
SoftminFuncOptions(int64_t dim)¶