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 (Any) –

  • y (Any) –

Return type

Any