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