jax.scipy.special.expit

jax.scipy.special.expit = <jax._src.custom_derivatives.custom_jvp object>[source]

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

LAX-backend implementation of 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.

Returns

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

Return type

ndarray