cupy.in1d#

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

测试一维数组的每个元素是否存在于第二个数组中。

返回一个与 ar1 长度相同的布尔数组，当 ar1 的元素存在于 ar2 中时，对应位置为 True，否则为 False。

参数:

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

  • ar2 (cupy.ndarray) – 用于测试 ar1 中每个值的数组。

  • assume_unique (bool, 可选) – 忽略

  • invert (bool, 可选) – 如果为 True，则返回数组中的值会反转（即，ar1 的元素在 ar2 中时为 False，否则为 True）。默认为 False。

返回:

y – 布尔数组，表示 ar1 中的哪些元素存在于 ar2 中。

返回类型:

cupy.ndarray, bool