jax.numpy.setxor1d

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

計算兩個陣列中元素的集合互斥或。

numpy.setxor1d() 的 JAX 實作。

由於 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 填補剩餘項目。fill_value 預設為互斥或結果中的最小值。

傳回：

在正好一個輸入陣列中找到的值陣列。

回傳型別：

Array

另請參閱

• jax.numpy.intersect1d()：兩個 1D 陣列的集合交集。

• jax.numpy.union1d()：兩個 1D 陣列的集合聯集。

• jax.numpy.setdiff1d()：兩個 1D 陣列的集合差集。

範例

>>> 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)