torch.topk() 函数详解

作用：返回列表中最大的n个值

例子1：m=torch.arange(0,10)
print(m.topk(3))

torch.return_types.topk(
values=tensor([9, 8, 7]),
indices=tensor([9, 8, 7]))

例子2：pred = torch.tensor([[-0.5816, -0.3873, -1.0215, -1.0145, 0.4053],
[ 0.7265, 1.4164, 1.3443, 1.2035, 1.8823],
[-0.4451, 0.1673, 1.2590, -2.0757, 1.7255],
[ 0.2021, 0.3041, 0.1383, 0.3849, -1.6311]])

values, indices = pred.topk(4, dim=0, largest=True, sorted=True)

print(values)
print(indices)

tensor([[ 0.7265,  1.4164,  1.3443,  1.2035,  1.8823],[ 0.2021,  0.3041,  1.2590,  0.3849,  1.7255],[-0.4451,  0.1673,  0.1383, -1.0145,  0.4053],[-0.5816, -0.3873, -1.0215, -2.0757, -1.6311]])
tensor([[1, 1, 1, 1, 1],[3, 3, 2, 3, 2],[2, 2, 3, 0, 0],[0, 0, 0, 2, 3]])

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None)

input -> 输入tensor
k -> 前k个
dim -> 默认为输入tensor的最后一个维度
sorted -> 是否排序
largest -> False表示返回第k个最小值

torch.topk() 函数详解相关推荐

Torch.arange函数详解
torch.arange函数详解官方文档:torch.arange 函数原型 arange(start=0, end, step=1, *, out=None, dtype=None, layout ...
【Pytorch】torch.argmax 函数详解
文章目录一.一个参数时的 torch.argmax 函数 1. 介绍 2. 实例二.多个参数时的 torch.argmax 函数 1. 介绍 2. 实例实例1:二维矩阵实例2:三维矩阵实例3 ...
torch.flatten()函数详解
自己的理解: 介绍torch.flatten()函数的具体使用方法1.首先创建一个三维张量2.调用torch.flatten()函数 import torchx = torch.randn(2, 3, ...
PyTorch中的topk函数详解
听名字就知道这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index. 用法 torch.topk(input, k, dim=None, largest=True, sor ...
【Torch API】pytorch 中torch.ones_like和torch.zeros_like函数详解
torch.ones_like函数和torch.zeros_like函数的基本功能是根据给定张量,生成与其形状相同的全1张量或全0张量,示例如下: input = torch.rand(2, 3) p ...
torch.zeros() 函数详解
torch.zeros()函数返回一个形状为为size,类型为torch.dtype,里面的每一个值都是0的tensor torch.zeros(*size, out=None, dtype=Non ...
torch.scatter函数详解
#torch.scatter函数官方解释 scatter(output, dim, index, src) → Tensor Writes all values from the tensor src ...
PyTorch中torch.norm函数详解
torch.norm() 是 PyTorch 中的一个函数,用于计算输入张量沿指定维度的范数.具体而言,当给定一个输入张量 x 和一个整数 p 时,torch.norm(x, p) 将返回输入张量 x ...
pytorch之torch.nn.Conv2d()函数详解
文章目录一.官方文档介绍二.torch.nn.Conv2d()函数详解参数详解参数dilation--扩张卷积(也叫空洞卷积) 参数groups--分组卷积三.代码实例一.官方文档介绍官 ...

torch.topk() 函数详解

torch.topk() 函数详解相关推荐

最新文章

热门文章