jax.nn.softplus

Contents

jax.nn.softplus#

jax.nn.softplus(x)[source]#

Softplus activation function.

Computes the element-wise function

\[\mathrm{softplus}(x) = \log(1 + e^x)\]
Parameters:

x (jax.typing.ArrayLike) – input array

Return type:

Array