cupy.digitize#
- cupy.digitize(x, bins, right=False)[源代码]#
查找输入数组中每个值所属的bin(桶)的索引。
注意
为了避免设备同步,当数组不是单调时,digitize 不会引发异常
- 参数:
x (cupy.ndarray) – 输入数组。
bins (cupy.ndarray) – bin(桶)数组。它必须是1维的,并且单调递增或递减。
right (bool) – 指示区间是否包含右边缘或左边缘。
- 返回:
输出索引数组,形状与
x
相同。- 返回类型:
另请参阅