cupy.digitize#

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

查找输入数组中每个值所属的箱子的索引。

备注

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

参数:

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

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

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

返回:

输出一个索引数组，其形状与 x 相同。

返回类型:

cupy.ndarray

参见

numpy.digitize()