jax.lax.conj

jax.lax.conj(x)

Elementwise complex conjugate function: \(\overline{x}\).

This function lowers to a combination of stablehlo.real, stablehlo.imag, and stablehlo.complex.

Parameters:

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

Returns:

Array of the same shape and dtype as x containing its complex conjugate.

Return type:

Array

See also

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

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

• jax.lax.imag(): elementwise extract imaginary part.

• jax.lax.abs(): elementwise absolute value / complex magnitude.