jax.numpy.iscomplex

jax.numpy.iscomplex(x)[source]

Returns a bool array, where True if input element is complex.

LAX-backend implementation of iscomplex(). Original docstring below.

What is tested is whether the input has a non-zero imaginary part, not if the input type is complex.

Parameters

x (array_like) – Input array.

Returns

out – Output array.

Return type

ndarray of bools

See also

isreal()

iscomplexobj()

Return True if x is a complex type or an array of complex numbers.

Examples

>>> np.iscomplex([1+1j, 1+0j, 4.5, 3, 2, 2j])
array([ True, False, False, False, False,  True])