jax.lax.exp2

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

Elementwise base-2 exponential: \(2^x\).

This function is implemented in terms of the stablehlo.exponential and stablehlo.multiply operations.

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 base-2 exponential.

Return type:

Array

See also

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

• jax.lax.log(): elementwise natural logarithm: \(\mathrm{log}(x)\).