jax.numpy.setxor1d

目录

jax.numpy.setxor1d#

jax.numpy.setxor1d(ar1, ar2, assume_unique=False, *, size=None, fill_value=None)[source]#

计算两个数组中元素的集合异或运算。

JAX 实现的 numpy.setxor1d()。

由于 setxor1d 输出的大小取决于数据，因此该函数与 JIT 或其他 JAX 变换不兼容。

参数：:

  • ar1 (ArrayLike) – 要相交的第一个值数组。

  • ar2 (ArrayLike) – 要相交的第二个值数组。

  • assume_unique (bool) – 如果为 True，则假设输入数组包含唯一值。这允许更有效的实现，但如果 assume_unique 为 True 且输入数组包含重复值，则行为未定义。默认值：False。

  • size (int | None | None) – 如果指定，则仅返回前 size 个排序的元素。如果元素数量少于 size 指示的数量，则返回值将用 fill_value 填充，返回的索引将用超出范围的索引填充。

  • fill_value (ArrayLike | None | None) – 当 size 指定且元素数量少于指示的数量时，用 fill_value 填充剩余条目。默认为异或结果中的最小值。

返回值:

一个包含在输入数组中只有一个的值的数组。

返回类型:

数组

另请参见

示例

>>> ar1 = jnp.array([1, 2, 3, 4])
>>> ar2 = jnp.array([3, 4, 5, 6])
>>> jnp.setxor1d(ar1, ar2)
Array([1, 2, 5, 6], dtype=int32)
内容