jax.numpy.all#

jax.numpy.all(a, axis=None, out=None, keepdims=False, *, where=None)[原始碼]#

測試沿著給定軸的所有陣列元素是否評估為 True。

JAX 實作的 numpy.all()。

參數:

  • a (ArrayLike) – 輸入陣列。

  • axis (Axis | None) – 整數或陣列，預設值=None。要沿著測試的軸。如果為 None，則沿著所有軸測試。

  • keepdims (bool) – bool，預設值=False。如果為 true，則縮減的軸會保留在結果中，大小為 1。

  • where (ArrayLike | None | None) – 整數或布林資料類型的陣列，預設值=None。要在測試中使用的元素。陣列應與輸入廣播相容。

  • out (None | None) – JAX 未使用。

傳回值:

布林值陣列。

傳回型別:

Array

範例

預設情況下，jnp.all 測試沿著所有軸的 True 值。

>>> x = jnp.array([[True, True, True, False],
...                [True, False, True, False],
...                [True, True, False, False]])
>>> jnp.all(x)
Array(False, dtype=bool)

如果 axis=0，則測試沿著軸 0 的 True 值。

>>> jnp.all(x, axis=0)
Array([ True, False, False, False], dtype=bool)

如果 keepdims=True，則輸出的 ndim 將與輸入的相同。

>>> jnp.all(x, axis=0, keepdims=True)
Array([[ True, False, False, False]], dtype=bool)

若要在測試 True 值時包含特定元素，您可以使用 ``where``。

>>> where=jnp.array([[1, 0, 1, 0],
...                  [0, 0, 1, 1],
...                  [1, 1, 1, 0]], dtype=bool)
>>> jnp.all(x, axis=0, keepdims=True, where=where)
Array([[ True,  True, False, False]], dtype=bool)