jax.lax.conj#

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

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

This function lowers to a combination ofstablehlo.real,stablehlo.imag,andstablehlo.complex.

Parameters:

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

Returns:

Array of the same shape and dtype asx 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.