Mila::Dnn::SoftmaxConfig Class Reference

Configuration class for Softmax module. More...

Inheritance diagram for Mila::Dnn::SoftmaxConfig:

Public Member Functions
void	fromMetadata (const SerializationMetadata &meta) override
	Deserialize configuration from SerializationMetadata.
int64_t	getAxis () const noexcept
	Get the configured axis value.
SerializationMetadata	toMetadata () const override
	Serialize this configuration to SerializationMetadata.
std::string	toString () const override
	Get a string representation of the configuration.
void	validate () const override
	Validate configuration parameters.
template<typename Self>
Self &&	withAxis (this Self &&self, int64_t axis)
	fluent setter for the axis along which to apply softmax.
Public Member Functions inherited from Mila::Dnn::ComponentConfig
virtual	~ComponentConfig ()=default
	Virtual destructor for polymorphic base.

Detailed Description

Configuration class for Softmax module.

Provides a type-safe fluent interface for configuring Softmax modules.

Member Function Documentation

fromMetadata()

void Mila::Dnn::SoftmaxConfig::fromMetadata ( const SerializationMetadata & meta )

inlineoverridevirtual

Deserialize configuration from SerializationMetadata.

Missing keys leave fields at their current values.

Parameters

meta	SerializationMetadata containing configuration parameters

Implements Mila::Dnn::ComponentConfig.

getAxis()

int64_t Mila::Dnn::SoftmaxConfig::getAxis ( ) const

inlinenoexcept

Get the configured axis value.

Returns

int64_t The axis along which softmax will be computed

toMetadata()

SerializationMetadata Mila::Dnn::SoftmaxConfig::toMetadata ( ) const

inlineoverridevirtual

Serialize this configuration to SerializationMetadata.

Returns

SerializationMetadata containing configuration parameters

Implements Mila::Dnn::ComponentConfig.

toString()

std::string Mila::Dnn::SoftmaxConfig::toString ( ) const

inlineoverridevirtual

Get a string representation of the configuration.

Returns

std::string Human-readable representation

Implements Mila::Dnn::ComponentConfig.

validate()

void Mila::Dnn::SoftmaxConfig::validate ( ) const

inlineoverridevirtual

Validate configuration parameters.

Exceptions

std::invalid_argument

If validation fails

Implements Mila::Dnn::ComponentConfig.

withAxis()

template<typename Self>

Self && Mila::Dnn::SoftmaxConfig::withAxis ( this Self && self, int64_t axis )

inline

fluent setter for the axis along which to apply softmax.

Parameters

axis	Dimension for softmax computation (default: -1 for last dimension)

Returns

Self&& for method chaining

---

The documentation for this class was generated from the following file:

• Mila/Src/Dnn/Components/Normalization/SoftmaxConfig.ixx