ott.neural.networks.potentials.PotentialMLP.act_fn

PotentialMLP.act_fn(negative_slope=0.01)

Leaky rectified linear unit activation function.

Computes the element-wise function:

\[\begin{split}\mathrm{leaky\_relu}(x) = \begin{cases} x, & x \ge 0\\ \alpha x, & x < 0 \end{cases}\end{split}\]

where \(\alpha\) = negative_slope.

Parameters:

• x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – input array

• negative_slope (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – array or scalar specifying the negative slope (default: 0.01)

Return type:

Array

Returns:

An array.

See also

relu()