jax.numpy.bitwise_and#

jax.numpy.bitwise_and = <jnp.ufunc 'bitwise_and'>#

逐元素計算位元 AND 運算。

numpy.bitwise_and 的 JAX 實作。這是一個通用函數，並支援 jax.numpy.ufunc 中描述的額外 API。此函數提供 JAX 陣列的 & 運算子實作。

參數：

  • x – 整數或布林陣列。必須可廣播到通用形狀。

  • y – 整數或布林陣列。必須可廣播到通用形狀。

  • args (ArrayLike)

  • out (None)

  • where (None)

回傳：

包含逐元素位元 AND 結果的陣列。

回傳類型：

Any

範例

明確呼叫 bitwise_and

>>> x = jnp.arange(4)
>>> jnp.bitwise_and(x, 1)
Array([0, 1, 0, 1], dtype=int32)

透過 & 運算子呼叫 bitwise_and

>>> x & 1
Array([0, 1, 0, 1], dtype=int32)