jax.numpy.array_equiv

jax.numpy.array_equiv(a1, a2)

检查两个数组是否逐元素相等。

JAX 实现的 numpy.array_equiv()。

如果输入数组不能广播到相同的形状，此函数将返回 False。

参数:

• a1 (ArrayLike) – 要比较的第一个输入数组。

• a2 (ArrayLike) – 第二个输入数组进行比较。

返回:

布尔标量数组，指示输入数组在广播后是否逐元素相等。

返回类型:

Array

参见

• jax.numpy.allclose()

• jax.numpy.array_equal()

示例

>>> jnp.array_equiv(jnp.array([1, 2, 3]), jnp.array([1, 2, 3]))
Array(True, dtype=bool)
>>> jnp.array_equiv(jnp.array([1, 2, 3]), jnp.array([1, 2, 4]))
Array(False, dtype=bool)
>>> jnp.array_equiv(jnp.array([[1, 2, 3], [1, 2, 3]]),
...                 jnp.array([1, 2, 3]))
Array(True, dtype=bool)