jax.nn.elu

Warning

This page was created from a pull request (#9655).

jax.nn.elu¶

jax.nn.elu(x, alpha=1.0)[source]¶

Exponential linear unit activation function.

Computes the element-wise function:

\[\begin{split}\mathrm{elu}(x) = \begin{cases} x, & x > 0\\ \alpha \left(\exp(x) - 1\right), & x \le 0 \end{cases}\end{split}\]
Parameters

  • x (Any) – input array

  • alpha (Any) – scalar or array of alpha values (default: 1.0)

Return type

Any

previous

jax.nn.hard_tanh

next

jax.nn.celu