olm.nn.activations.base¶

Classes¶

`ActivationBase`(args, *kwargs)	Abstract base class for all activation functions.

class olm.nn.activations.base.ABC¶

Bases: object

Helper class that provides a standard way to create an ABC using inheritance.

class olm.nn.activations.base.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.

olm.nn.activations.base.abstractmethod(funcobj)¶

A decorator indicating abstract methods.

Requires that the metaclass is ABCMeta or derived from it. A class that has a metaclass derived from ABCMeta cannot be instantiated unless all of its abstract methods are overridden. The abstract methods can be called using any of the normal ‘super’ call mechanisms. abstractmethod() may be used to declare abstract methods for properties and descriptors.

Usage:

class C(metaclass=ABCMeta): : @abstractmethod def my_abstract_method(self, arg1, arg2, argN):

…