cupy.isin#

cupy.isin(element, test_elements, assume_unique=False, invert=False)[source]#

计算 element 中的元素是否在 test_elements 中，仅对 element 进行广播。返回一个与 element 形状相同的布尔数组，其中 element 的元素存在于 test_elements 中时为 True，否则为 False。

参数:

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

  • test_elements (cupy.ndarray) – 用于测试 element 中每个值的参考值。如果此参数是数组或类数组，则会被展平。

  • assume_unique (bool, optional) – 被忽略

  • invert (bool, optional) – 如果为 True，则返回数组中的值被反转，就像计算 element 的元素不在 test_elements 中一样。默认为 False。

返回:

y – 与 element 的形状相同。element[isin] 的值存在于 test_elements 中。

返回类型:

cupy.ndarray, bool