jax.lax.logistic

Contents

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.

  • accuracy (Tolerance | AccuracyMode | None) – Optional lax.Tolerance or lax.AccuracyMode object that selects the implementation of the op based on the requested accuracy. If the implementation cannot satisfy the requested tolerance, the compiler will return an error. If mode is specified and there are no multiple implementations available, the default implementation will be used.

Returns:

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

Return type:

Array

See also

Contents