jax.numpy.any#

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

測試沿著指定軸的任何陣列元素是否評估為 True。

JAX 實作的 numpy.any()。

參數:

  • a (ArrayLike) – 輸入陣列。

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

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

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

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

返回:

布林值的陣列。

返回類型:

Array

範例

預設情況下，jnp.any 會沿著所有軸進行測試。

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

如果 axis=0，則沿著軸 0 進行測試。

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

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

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

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

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