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)