jax.scipy.special.expit

Contents

jax.scipy.special.expit#

jax.scipy.special.expit(x)[source]#

The logistic sigmoid (expit) function

JAX implementation of scipy.special.expit.

\[\mathrm{expit}(x) = \frac{1}{1 + e^{-x}}\]
Parameters:

x (jax.typing.ArrayLike) – arraylike, real-valued.

Returns:

array containing values of the expit function.

Return type:

Array