Class Softmax

• Defined in File activation.h

Inheritance Relationships

Base Type

• public torch::nn::ModuleHolder< SoftmaxImpl > (Template Class ModuleHolder)

Class Documentation

class Softmax : public torch::nn::ModuleHolder<SoftmaxImpl>

A ModuleHolder subclass for SoftmaxImpl.

See the documentation for SoftmaxImpl class to learn what methods it provides, and examples of how to use Softmax with torch::nn::SoftmaxOptions. See the documentation for ModuleHolder to learn about PyTorch’s module storage semantics.

Public Types

using Impl = SoftmaxImpl