分类：topk函数

2021
07-04

使用numpy实现topk函数操作(并排序)

np.argpartition难以解决topKtopK是常用的一个功能，在python中，numpy等计算库使用了丰富的底层优化，对于矩阵计算的效率远高于python的for-loop实现。因此，我们希望尽量用一些numpy函数的组合实现topK。pytorch库提供了topk函数，可以将高维数组沿某一维度（该维度共N项），选出最大（最小）的K项并排序。返回排序结果和index信息。奇怪的是，更轻量级的numpy库并没有直接提供topK函数。numpy只提供了argpartition和parti... 继续阅读 >

2020
09-25

python topk()函数求最大和最小值实例

函数介绍a.topk()求a中的最大值或最小值，返回两个值，一个是a中的值（最大或最小），一个是这个值的索引。代码示例>>>importtorch>>>a=torch.randn((3,5))>>>atensor([[-0.4790,-0.6308,0.2370,0.0380,-0.0579],[-0.6712,-3.5483,-0.2370,-0.8658,0.4145],[-1.4126,-0.8786,-0.4216,-0.0878,-1.4015]])>>>_,pre=a.topk(1,dim=1,largest=True)>>>pretensor([[2],[4],[3]])>>>_tensor([[0.2370],[0.... 继续阅读 >