jax.numpy.iscomplex

jax.numpy.iscomplex(x)

Return boolean array showing where the input is complex.

JAX implementation of numpy.iscomplex().

Parameters:

x (ArrayLike) – Input array to check.

Returns:

A new array containing boolean values indicating complex elements.

Return type:

Array

See also

• jax.numpy.iscomplexobj()

• jax.numpy.isrealobj()

Examples

>>> jnp.iscomplex(jnp.array([True, 0, 1, 2j, 1+2j]))
Array([False, False, False, True, True], dtype=bool)