jax.lax.imag#

jax.lax.imag(x)[source]#

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

This function lowers directly to thestablehlo.imag operation.

Parameters:

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

Returns:

Array of the same shape asx containing its imaginary part. Will have dtypefloat32 ifx.dtype==complex64, or float64 ifx.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.