jax.numpy.bitwise_xor#
- jax.numpy.bitwise_xor = <jnp.ufunc 'bitwise_xor'>#
逐元素计算按位异或操作。
JAX 对
numpy.bitwise_xor
的实现。这是一个通用函数,支持在jax.numpy.ufunc
中描述的额外 API。此函数提供了 JAX 数组的^
运算符实现。- 参数:
x – 整数或布尔数组。必须能够广播到共同的形状。
y – 整数或布尔数组。必须能够广播到共同的形状。
args (ArrayLike)
out (None)
where (None)
- 返回:
包含逐元素按位异或结果的数组。
- 返回类型:
任意类型
示例
显式调用
bitwise_xor
>>> x = jnp.arange(4) >>> jnp.bitwise_xor(x, 1) Array([1, 0, 3, 2], dtype=int32)
通过
^
运算符调用bitwise_xor
>>> x ^ 1 Array([1, 0, 3, 2], dtype=int32)