问从没有重复的张量中选择顶K值
EN

Stack Overflow用户

提问于 2021-07-30 09:41:03

回答 1查看 427关注 0票数 0

torch.Tensor.topk提供了一种在一维张量中提取顶k值的有效方法。是否可以将顶部k值限制为non-repetitive

例如,

input = torch.tensor([0.2,0.2,0.1])
k = 2
dim = 0


output[0] = torch.tensor([0.2,0.1])
output[1] = torch.longtensor([0,2])

发布于 2021-07-30 09:53:49

您可以在输入张量上应用torch.unique。

>>> input.unique().topk(k=2).values
tensor([0.2000, 0.1000])

请注意，此时你将失去指数。

编辑:实际上，torch.unique有一个选项来对结果进行排序(默认情况下该选项是打开的)。

>>> input
tensor([0.0000, 0.3000, 0.2000, 0.2000, 0.1000])

>>> input.unique(return_inverse=True)[1].unique(sorted=False)
tensor([1, 2, 3, 0])

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/68588784

复制

相似问题

问从没有重复的张量中选择顶K值EN