cupy.digitize#

cupy.digitize(x, bins, right=False)[源代码]#

查找输入数组中每个值所属的bin(桶)的索引。

注意

为了避免设备同步,当数组不是单调时,digitize 不会引发异常

参数:
  • x (cupy.ndarray) – 输入数组。

  • bins (cupy.ndarray) – bin(桶)数组。它必须是1维的,并且单调递增或递减。

  • right (bool) – 指示区间是否包含右边缘或左边缘。

返回:

输出索引数组,形状与 x 相同。

返回类型:

cupy.ndarray

另请参阅

numpy.digitize()