jax.scipy.special.expit

Contents

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