cupy.setdiff1d#

cupy.setdiff1d(ar1, ar2, assume_unique=False)[source]#

查找两个数组的集合差集。它返回 ar1 中不在 ar2 中的唯一值。

参数：

  • ar1 (cupy.ndarray) – 输入数组

  • ar2 (cupy.ndarray) – 用于比较的输入数组

  • assume_unique (bool) – 默认为 False，即输入数组不保证唯一。如果为 True，则假定输入数组是唯一的。这可以加快计算速度。

返回值：

setdiff1d – 返回一个一维数组，包含 ar1 中不在 ar2 中的值。对于未排序的输入，只有在 assume_unique=False 时，返回结果才保证有序。

返回类型：

cupy.ndarray

另请参阅

numpy.setdiff1d