jax.numpy.digitize#
- jax.numpy.digitize(x, bins, right=False)[source]#
返回输入数组中每个值所属的 bin 的索引。
LAX 后端实现的
numpy.digitize().原始文档字符串如下。
right
bin 的顺序
返回的索引 i 满足
False递增
bins[i-1] <= x < bins[i]True递增
bins[i-1] < x <= bins[i]False递减
bins[i-1] > x >= bins[i]True递减
bins[i-1] >= x > bins[i]如果 x 中的值超出 bins 的范围，则根据情况返回 0 或
len(bins)。- 参数：
x (array_like) – 要进行 binning 的输入数组。在 NumPy 1.10.0 之前，此数组必须是一维的，但现在可以具有任何形状。
bins (array_like) – bin 数组。它必须是一维的且单调的。
right (bool, optional) – 指示区间是否包含右边界或左边界。默认行为是 (right==False)，表示区间不包含右边界。在这种情况下，左 bin 端是开放的，即 bins[i-1] <= x < bins[i] 是单调递增 bin 的默认行为。
- 返回值：
indices – 索引的输出数组，与 x 形状相同。
- 返回类型：
整数 ndarray