jax.numpy.iscomplex

Contents

jax.numpy.iscomplex#

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

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

LAX-backend implementation of numpy.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