jax.numpy.bitwise_and

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

计算按位与运算的元素。

numpy.bitwise_and 的 JAX 实现。这是一个通用函数，并且支持 jax.numpy.ufunc 中描述的其他 API。此函数为 JAX 数组提供了 & 运算符的实现。

参数：

• x – 整数或布尔数组。必须可广播到公共形状。

• y – 整数或布尔数组。必须可广播到公共形状。

• args (ArrayLike)

• out (None)

• where (None)

返回：

包含按元素进行按位与运算结果的数组。

返回类型：

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)