jax.nn.soft_sign

jax.nn.soft_sign(x)[source]

Soft-sign activation function.

Computes the element-wise function

\[\mathrm{soft\_sign}(x) = \frac{x}{|x| + 1}\]
Parameters

x (Any) – input array

Return type

Any