Skip to main content
Ctrl+K

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 (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

  • y (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

Return type:

Array

previous

jax.lax.collapse

next

jax.lax.concatenate