jax.lax.complex

Contents

jax.lax.complex#

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

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

Builds a complex number from real and imaginary parts.

Parameters:

x (jax.typing.ArrayLike)
y (jax.typing.ArrayLike)

Return type: