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 时提供。
另请参阅