jax.numpy.iscomplex

jax.numpy.iscomplex(x)

返回布尔数组，显示输入为复数的位置。

JAX 对 numpy.iscomplex() 的实现。

参数:

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

返回值:

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

返回类型:

数组

另请参见

• jax.numpy.iscomplexobj()

• jax.numpy.isrealobj()

示例

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