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 (ArrayLike)

  • y (ArrayLike)

Return type:

Array