olm.nn.activations.selu¶
Classes¶
SELU(*args, **kwargs) |
SELU activation wrapper. |
|---|---|
class olm.nn.activations.selu.ActivationBase(*args: Any, **kwargs: Any)¶
Bases: Module, ABC
Abstract base class for all activation functions.
Ensures a consistent interface for activation layers, handling device and dtype initialization. Subclasses must implement the forward method.
device¶
The device the module is on.
- Type: torch.device, optional
dtype¶
The data type of the module parameters.
- Type: torch.dtype
abstractmethod forward(x: torch.Tensor) → torch.Tensor¶
Apply activation to x.
class olm.nn.activations.selu.SELU(*args: Any, **kwargs: Any)¶
Bases: ActivationBase
SELU activation wrapper.
forward(x: torch.Tensor) → torch.Tensor¶
Apply activation to x.