pytorch 中bincount()函数详解

不当菜鸡的程序媛2024-05-31 11:40

torch.bincount 函数通常用于统计离散值的出现次数，如图中节点的批次信息、类别标签等。它可以帮助我们快速计算每个值的计数，而无需手动编写循环或其他复杂的逻辑。

例子：

复制代码

>>> import torch
>>> input = torch.tensor([1, 2, 2, 3, 3, 3, 10])
>>> counts = torch.bincount(input)
>>> print(counts)
tensor([0, 1, 2, 3, 0, 0, 0, 0, 0, 0, 1])

tensor([0, 1, 2, 3, 0, 0, 0, 0, 0, 0, 1])。计算统计的频次，即，0对应的标签0个，以此类推，1:1个，2:2个，3:3个，4:0个，5:0个，6:0个，7:0个，8:0个，9:0个，10:1个。