pytorch中tensor.topk

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

topk的原型如上：

其中k是保留的k个值，largest=True意味着选取最大的，sorted=True是指将返回结果排序

topk返回的是一个tuple，第一个元素指返回的具体值，第二个元素指返回值的index

直接贴代码

import torchx = torch.rand(2, 3, 3)
y = x.topk(2, largest = True, sorted = True)

结果如下

>>> x
tensor([[[0.0858, 0.4492, 0.4394],[0.2662, 0.5704, 0.5212],[0.1720, 0.8962, 0.4634]],[[0.5535, 0.5748, 0.0194],[0.0723, 0.7901, 0.4427],[0.7804, 0.8924, 0.3323]]])>>> y
(tensor([[[0.4492, 0.4394],[0.5704, 0.5212],[0.8962, 0.4634]],[[0.5748, 0.5535],[0.7901, 0.4427],[0.8924, 0.7804]]]), tensor([[[1, 2],[1, 2],[1, 2]],[[1, 0],[1, 2],[1, 0]]]))

可以看到sorted的作用，是在返回的tuple中，将返回值也排序。

dim的作用，在指定的维度进行取最大最小

如一个矩阵

[0.0858, 0.4492, 0.4394]
[0.2662, 0.5704, 0.5212]
[0.1720, 0.8962, 0.4634]

dim=1是指在列维度进行取最大，也就是寻找每一行的最大值

dim=0是指在行维度取进行最大，也就是寻找每一列的最大值（有点跟逻辑反着的意思）

pytorch中tensor.topk相关推荐

PyTorch中的topk方法以及分类Top-K准确率的实现
PyTorch中的topk方法以及分类Top-K准确率的实现 Top-K 准确率在分类任务中的类别数很多时(如ImageNet中1000类),通常任务是比较困难的,有时模型虽然不能准确地将groun ...
pytorch tensor查找0_在PyTorch中Tensor的查找和筛选例子
本文源码基于版本1.0,交互界面基于0.4.1 import torch 按照指定轴上的坐标进行过滤 index_select() 沿着某tensor的一个轴dim筛选若干个坐标 >>&g ...
Pytorch中tensor.view().permute().contiguous()函数理解
Pytorch中tensor.view().permute().contiguous()函数理解 yolov3中有一行这样的代码,在此记录一下三个函数的含义 # 例子中batch_size为整型,le ...
Pytorch中tensor.expand()和tensor.expand_as()函数
Pytorch中tensor.expand函数 Tensor.expand()函数详解 Tensor.expand_as()函数 Tensor.expand()函数详解函数语法: # 官方解释: D ...
Pytorch中tensor维度和torch.max()函数中dim参数的理解
Pytorch中tensor维度和torch.max()函数中dim参数的理解维度参考了 https://blog.csdn.net/qq_41375609/article/details/106 ...
Pytorch中Tensor的索引，切片以及花式索引(fancy indexing)
目录理解Tensor的dim 索引简单索引用1维的list,numpy,tensor索引用booltensor索引切片花式索引结语前一段时间遇到一个花式索引的问题,在搜索良久之后没有找 ...
pytorch 中的topk函数
pytorch中topk() 函数用法 1. 函数介绍最近在代码中看到这两个语句 maxk = max(topk) _, pred = output.topk(maxk, 1, True, True ...
pyTorch中tensor运算
文章目录 PyTorch的简介 PyTorch中主要的包 PyTorch的安装使用GPU的原因使数据在GPU上运行什么使Tensor(张量) 一些术语介绍 Tensor的属性介绍(Rank,ax ...
PyTorch中tensor介绍
PyTorch中的张量(Tensor)如同数组和矩阵一样,是一种特殊的数据结构.在PyTorch中,神经网络的输入.输出以及网络的参数等数据,都是使用张量来进行描述. torc ...
pytorch中tensor、backward一些总结
目录说明 Tensor Tensor的创建 Tensor(张量)基本数据类型与常用属性 Tensor的自动微分设置不可积分计算 pytorch 计算图 backward一些细节该文章解决问题如下 ...

pytorch中tensor.topk

pytorch中tensor.topk相关推荐

最新文章

热门文章