jax.lax.logistic#

jax.lax.logistic(x, accuracy=None)[source]#

Elementwise logistic (sigmoid) function: \(\frac{1}{1 + e^{-x}}\).

There is no HLO logistic/sigmoid primitive, so this lowers to a sequence of HLO arithmetic operations.

Parameters:

x (ArrayLike) – input array. Must have floating point or complex dtype.

Returns:

Array of the same shape and dtype as x containing the element-wise logistic/sigmoid function.

Return type:

Array

See also