jax.numpy.bitwise_and#
- jax.numpy.bitwise_and = <jnp.ufunc 'bitwise_and'>#
逐元素计算按位 AND 运算。
JAX 实现的
numpy.bitwise_and
。这是一个通用函数,并支持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)