jax.nn.soft_sign

Warning

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

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

previous

jax.nn.softplus

next

jax.nn.silu