cupy.intersect1d#

cupy.intersect1d(arr1, arr2, assume_unique=False, return_indices=False)[源代码][源代码]#

查找两个数组的交集。返回在两个输入数组中都存在的已排序且唯一的值。

参数:

  • arr1 (cupy.ndarray) – 输入数组。如果数组不是一维的，它们将被展平。

  • arr2 (cupy.ndarray) – 输入数组。如果数组不是一维的，它们将被展平。

  • assume_unique (bool) – 默认情况下，False。如果设置为True，输入数组将被假定为唯一的，这将加快计算速度。如果设置为True，但数组不是唯一的，可能会导致结果不正确和索引越界。

  • return_indices (bool) – 默认情况下，False。如果为True，则返回对应于两个数组交集的索引。

返回:

  • intersect1d (cupy.ndarray) – 包含共同和唯一元素的排序一维数组。

  • comm1 (cupy.ndarray) – arr1 中共同值首次出现的索引。仅在 return_indices 为 True 时提供。

  • comm2 (cupy.ndarray) – arr2 中共同值首次出现的索引。仅在 return_indices 为 True 时提供。

参见

numpy.intersect1d