jax.nn.celu#
- jax.nn.celu(x, alpha=1.0)[source]#
Continuously-differentiable exponential linear unit activation.
Computes the element-wise function:
\[\begin{split}\mathrm{celu}(x) = \begin{cases} x, & x > 0\\ \alpha \left(\exp(\frac{x}{\alpha}) - 1\right), & x \le 0 \end{cases}\end{split}\]For more information, see Continuously Differentiable Exponential Linear Units.
- Parameters:
x (jax.typing.ArrayLike) – input array
alpha (jax.typing.ArrayLike) – array or scalar (default: 1.0)
- Returns:
An array.
- Return type: