Exponential Linear Unit (ELU)#

API#

class warp_nn.modules.activations.ELU(*, alpha: float = 1.0)[source]#

Bases: Module

Exponential Linear Unit (ELU) activation function.

This class computes the element-wise ELU activation function:

\[\begin{split}\text{ELU}(x) = \begin{cases} x, & \text{ if } x \geq 0\\ \alpha \, (e^x - 1), & \text{ if } x < 0 \end{cases}\end{split}\]
Parameters:

alpha – The alpha value for the ELU function.

__call__(input: array) → array[source]#

Forward pass of the activation function.

Parameters:

input – The input array, with up to 3 dimensions.

Returns:

The output array, with same shape as the input array.

property alpha[source]#

The alpha value for the ELU function.