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