jax.numpy.iscomplexobj

jax.numpy.iscomplexobj#

jax.numpy.iscomplexobj(x)[源代码][源代码]#

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

JAX 实现的 numpy.iscomplexobj()

该函数根据输入类型而不是值进行评估。即使输入的虚部为零,仍被视为复数。

参数:

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

返回:

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

返回类型:

bool

示例

>>> 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