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