jax.lax.expm1

jax.lax.expm1(x, accuracy=None)

Elementwise \(e^{x} - 1\).

This function lowers directly to the stablehlo.exponential_minus_one operation. Compared to the naive expression lax.exp(x) - 1, it is more accurate for x near zero.

Parameters:

• x (ArrayLike) – input array. Must have floating-point or complex type.

• accuracy – 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 exponential minus 1.

Return type:

Array

See also

• jax.lax.exp(): elementwise exponentional: \(e^x\).

• jax.lax.log1p(): elementwise \(\mathrm{log}(1 + x)\).