jax.experimental.pallas.atomic_xor

jax.experimental.pallas.atomic_xor(x_ref_or_view, idx, val, *, mask=None)

原子性地計算 x_ref_or_view[idx] ^= val。

參數:

• x_ref_or_view – 要操作的 ref。

• idx – 要使用的索引器。

• mask (Any | None | None) – 待撰寫文件。

傳回:

原子操作前，給定索引處的值。