jax.scipy.special.expit#

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

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