pytorch中torch.topk()函数的快速理解

函数作用：

该函数的作用即按字面意思理解，topk:取数组的前k个元素进行排序。

通常该函数返回2个值，第一个值为排序的数组，第二个值为该数组中获取到的元素在原数组中的位置标号。

举个栗子：

该函数功能经常用来获取张量或者数组中最大或者最小的元素以及索引位置，是一个经常用到的基本函数。

实例演示

任务一：

取top1（最大值）：

任务二：

按行取出topk，将小于topk的置为inf：

任务三：

总结

到此这篇关于pytorch中torch.topk()函数快速理解的文章就介绍到这了,更多相关pytorch torch.topk()函数理解内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持！