cupyx.jit.atomic_sub

cupyx.jit.atomic_sub(array, index, value, alt_value=None) = <cupyx.jit function>

调用 atomicSub 函数对 array[index] 执行原子操作。有关详细说明，请参阅 原子函数。

参数:

• array – 用于索引的 cupy.ndarray。

• index – 有效索引，可用于计算对应数组元素 array[index] 的地址。

• value – 表示用于指定操作的值。对于 atomic_cas 的情况，这是与 array[index] 进行比较的值。

• alt_value – 仅在 atomic_cas 中使用，表示要交换到的值。

另请参阅

Numba 中对应的原子函数