jax.lax.complex#

jax.lax.complex(x,y)[source]#

Elementwise make complex number:\(x + jy\).

This function lowers directly to thestablehlo.complex operation.

Parameters:

• x (ArrayLike) – input arrays. Must have matching floating-point dtypes. Ifneither is a scalar, the two arrays must have the same numberof dimensions and be broadcast-compatible.

• y (ArrayLike) – input arrays. Must have matching floating-point dtypes. Ifneither is a scalar, the two arrays must have the same numberof dimensions and be broadcast-compatible.

Returns:

The complex array with the real part given byx, and the imaginarypart given byy. For inputs of dtype float32 or float64, the resultwill have dtype complex64 or complex128 respectively.

Return type:

Array

See also

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

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

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