output.topk()函数解读

代码如下

>>> a=torch.randn((4,6))
>>> print(a)
tensor([[ 0.7042,  0.2533,  1.1596, -0.7436,  0.5264,  0.2085],[ 0.2641,  0.9683,  0.4469, -1.9215, -0.7564,  1.1776],[ 1.0520, -1.6003, -0.8634,  1.7596, -0.8464,  0.7166],[-0.0492, -0.7746,  1.2592, -0.8273,  0.1266,  1.0450]])
>>> maxk=max((1,3))
>>> _, pred=a.topk(maxk,1,True,True)
>>> print(_)
tensor([[1.1596, 0.7042, 0.5264],[1.1776, 0.9683, 0.4469],[1.7596, 1.0520, 0.7166],[1.2592, 1.0450, 0.1266]])
>>> print(pred)
tensor([[2, 0, 4],[5, 1, 2],[3, 0, 5],[2, 5, 4]])
>>> _, pred=a.topk(1,1,True,True)
>>> print(_)
tensor([[1.1596],[1.1776],[1.7596],[1.2592]])
>>> print(pred)
tensor([[2],[5],[3],[2]])

如上我们可以看到，topk()函数取指定维度上的最大值(或最大几个)，第二个参数dim=1，为按行取，dim=0，为按列取，如下：

>>> _, pred=a.topk(1,0,True,True)
>>> print(_)
tensor([[1.0520, 0.9683, 1.2592, 1.7596, 0.5264, 1.1776]])
>>> print(pred)
tensor([[2, 1, 3, 2, 0, 1]])

output.topk()函数解读相关推荐

pytorch 中的topk函数
pytorch中topk() 函数用法 1. 函数介绍最近在代码中看到这两个语句 maxk = max(topk) _, pred = output.topk(maxk, 1, True, True ...
relpos函数解读
relpos函数解读在配合网络的一些解读和一些自己的理解,将RTKLIB2.3.2的relpos函数进行解读流程: zdres-ddres-ekf求出浮点解----lambda固定解-----固定 ...
torch中的topk()函数
torch中的topk()函数 In [2]: import torchIn [3]: a=torch.randn((4,6))In [4]: a Out
硬阈值(Hard Thresholding)函数解读
版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明. 题目:硬阈值(Hard Thresholding)函数解读 1.硬阈值(Hard Thresh ...
PyTorch中的topk函数详解
听名字就知道这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index. 用法 torch.topk(input, k, dim=None, largest=True, sor ...
uip1.0核心模块uip_process函数解读
转载地址:https://www.amobbs.com/thread-5531817-1-1.html 最近,利用uip搞了小东西,要想利用好uip, 最好彻底搞清楚其关键函数 uip_process ...
软阈值函数作用图matlab,软阈值(Soft Thresholding)函数解读
题目:软阈值(Soft Thresholding) 函数解读 1.软阈值(Soft Thresholding)函数的符号软阈值(Soft Thresholding)目前非常常见,文献[1][2]最早 ...
软阈值（Soft Thresholding）函数解读
版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明. 题目:软阈值(Soft Thresholding) 函数解读 1.软阈值(Soft Thres ...
软阈值(Soft Thresholding) 函数解读
题目:软阈值(Soft Thresholding) 函数解读 1.软阈值(Soft Thresholding)函数的符号软阈值(Soft Thresholding)目前非常常见,文献[1][2]最早 ...

output.topk()函数解读

output.topk()函数解读相关推荐

最新文章

热门文章