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)