jax.numpy.iscomplexobj

jax.numpy.iscomplexobj(x)

Check if the input is a complex number or an array containing complex elements.

JAX implementation of numpy.iscomplexobj().

The function evaluates based on input type rather than value. Inputs with zero imaginary parts are still considered complex.

Parameters:

x (Any) – input object to check.

Returns:

True if x is a complex number or an array containing at least one complex element, False otherwise.

Return type:

bool

See also

• jax.numpy.isrealobj()

• jax.numpy.iscomplex()

Examples

>>> jnp.iscomplexobj(True)
False
>>> jnp.iscomplexobj(0)
False
>>> jnp.iscomplexobj(jnp.array([1, 2]))
False
>>> jnp.iscomplexobj(1+2j)
True
>>> jnp.iscomplexobj(jnp.array([0, 1+2j]))
True