SoftmaxConfig.ixx File Reference

Configuration interface for the Softmax module in the Mila DNN framework. More...

#include <stdexcept>
#include <cstdint>
import Dnn.ComponentConfig;
import Dnn.Module;

Classes
class	Mila::Dnn::SoftmaxConfig
	Configuration class for Softmax module. More...

Namespaces
namespace	Mila
namespace	Mila::Dnn

Detailed Description

Configuration interface for the Softmax module in the Mila DNN framework.

Defines the SoftmaxConfig class, providing a type-safe fluent interface for configuring Softmax activation function modules. Inherits from ComponentConfig CRTP base and adds Softmax-specific options such as the axis along which to apply the operation.

Exposed as part of the Softmax module via module partitions.