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
