cupy.setdiff1d#

cupy.setdiff1d(ar1, ar2, assume_unique=False)[源代码][源代码]#

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

参数:

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

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

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

返回:

setdiff1d – 返回一个包含在 ar1 中但不在 ar2 中的值的一维数组。仅当 assume_unique=False 时，对于未排序的输入，它总是返回排序后的输出。

返回类型:

cupy.ndarray

参见

numpy.setdiff1d