jax.numpy.isreal — JAX documentation (original) (raw)

jax.numpy.isreal#

jax.numpy.isreal(x)[source]#

Return boolean array showing where the input is real.

JAX implementation of numpy.isreal().

Parameters:

x (ArrayLike) – input array to check.

Returns:

A new array containing boolean values indicating real elements.

Return type:

Array

Examples

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