jax.numpy.iscomplexobj

jax.numpy.iscomplexobj(x)

检查输入是否为复数或包含复数元素的数组。

numpy.iscomplexobj() 的 JAX 实现。

此函数基于输入类型而非值进行评估。虚部为零的输入仍然被认为是复数。

参数:

x (Any) – 要检查的输入对象。

返回:

如果 x 是复数或包含至少一个复数元素的数组，则返回 True，否则返回 False。

返回类型:

bool

另请参阅

• jax.numpy.isrealobj()

• jax.numpy.iscomplex()

示例

>>> jnp.iscomplexobj(True)
False
>>> jnp.iscomplexobj(0)
False
>>> jnp.iscomplexobj(jnp.array([1, 2]))
False
>>> jnp.iscomplexobj(1+2j)
True
>>> jnp.iscomplexobj(jnp.array([0, 1+2j]))
True