jax.scipy.special.expit

jax.scipy.special.expit(x)

Expit (a.k.a. logistic sigmoid) ufunc for ndarrays.

LAX-backend implementation of scipy.special.expit().

Original docstring below.

The expit function, also known as the logistic sigmoid function, is defined as expit(x) = 1/(1+exp(-x)). It is the inverse of the logit function.

Parameters:

• x (ndarray) – The ndarray to apply expit to element-wise.

• out (ndarray, optional) – Optional output array for the function values

Returns:

An ndarray of the same shape as x. Its entries are expit of the corresponding entry of x.

Return type:

scalar or ndarray