jax.lax.imag

jax.lax.imag(x)

Elementwise extract imaginary part: \(\mathrm{Im}(x)\).

This function lowers directly to the stablehlo.imag operation.

Parameters:

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

Returns:

Array of the same shape as x containing its imaginary part. Will have dtype float32 if x.dtype == complex64, or float64 if x.dtype == complex128.

Return type:

Array

See also

• jax.lax.complex(): elementwise construct complex number.

• jax.lax.real(): elementwise extract real part.

• jax.lax.conj(): elementwise complex conjugate.