jax.numpy.iscomplex

jax.numpy.iscomplex(x)

返回一个布尔数组，显示输入中的哪些元素是复数。

JAX 对 numpy.iscomplex() 的实现。

参数:

x (ArrayLike) – 要检查的输入数组。

返回:

一个包含布尔值的新数组，指示哪些元素是复数。

返回类型:

Array

另请参阅

• jax.numpy.iscomplexobj()

• jax.numpy.isrealobj()

示例

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