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)

返回值:

包含按元素按位异或结果的数组。

返回类型:

Any

示例

显式调用 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)