快捷方式

torch.bucketize

torch.bucketize(input, boundaries, *, out_int32=False, right=False, out=None) Tensor

返回 input 中每个值的所属桶的索引,其中桶的边界由 boundaries 设置。返回一个与 input 大小相同的新的张量。如果 right 为 False(默认),则左边界是开放的。请注意,此行为与 numpy.digitize 的行为相反。更正式地说,返回的索引满足以下规则

right

返回的索引满足

False

boundaries[i-1] < input[m][n]...[l][x] <= boundaries[i]

True

boundaries[i-1] <= input[m][n]...[l][x] < boundaries[i]

参数
  • input (Tensor or Scalar) – 包含搜索值(s) 的 N 维张量或标量。

  • boundaries (Tensor) – 1 维张量,必须包含严格递增的序列,否则返回值未定义。

关键字参数
  • out_int32 (bool, 可选) – 指示输出数据类型。如果为 True,则为 torch.int32,否则为 torch.int64。默认值为 False,即默认输出数据类型为 torch.int64。

  • right (bool, 可选) – 如果为 False,则返回找到的第一个合适位置。如果为 True,则返回最后一个索引。如果没有找到合适的索引,则返回 0(对于非数值值(例如 nan、inf))或 boundaries 的大小(最后一个索引之后的一个)。换句话说,如果为 False,则从 boundaries 中获取 input 中每个值的较低边界索引。如果为 True,则获取上限索引。默认值为 False。

  • out (Tensor, 可选) – 输出张量,如果提供,则必须与 input 大小相同。

示例

>>> boundaries = torch.tensor([1, 3, 5, 7, 9])
>>> boundaries
tensor([1, 3, 5, 7, 9])
>>> v = torch.tensor([[3, 6, 9], [3, 6, 9]])
>>> v
tensor([[3, 6, 9],
        [3, 6, 9]])
>>> torch.bucketize(v, boundaries)
tensor([[1, 3, 4],
        [1, 3, 4]])
>>> torch.bucketize(v, boundaries, right=True)
tensor([[2, 3, 5],
        [2, 3, 5]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源