jax.nn.soft_sign

jax.nn.soft_sign(x)

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