Pytorch常用函数-topk

_腾飞

于 2022-06-02 22:20:29 发布

阅读量1.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： # 基础包文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/helloworld_Fly/article/details/125107484

39 篇文章

订阅专栏

本文详细介绍了如何在PyTorch中使用topk()函数，包括其基本原理、参数解释和实际操作示例。通过torch.topk函数，可以获取输入张量中指定维度的前k个最大或最小值及其对应的索引。这对于数据分析和排序任务非常实用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Topk

topk()方法用于返回输入数据中特定维度上的前k个最大的元素。

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

参数说明：

>>> x = torch.arange(1., 6.)
>>> x
tensor([1.,2.,3.,4.,5.])
>>> torch.topk(x, 3)
tensor([5.,4.,3.]), indices=tensor([4,3,2])

使用torch.topk可以指定特定维度、特定前N个最大/最小值，并输出其值和索引。