torch.argmax()函数

argmax函数：torch.argmax(input, dim=None, keepdim=False) 返回指定维度最大值的序号，dim给定的定义是：the demention to reduce，就是把dim这个维度，变成这个维度的最大值的index。

1）dim表示不同维度。特别的在dim=0表示二维矩阵中的列，dim=1在二维矩阵中的行。广泛的来说，我们不管一个矩阵是几维的，比如一个矩阵维度如下：(d0,d1,…,dn−1) ，那么dim=0就表示对应到d0 也就是第一个维度，dim=1表示对应到也就是第二个维度，以此类推。

2）知道dim的值是什么意思还不行，还要知道函数中这个dim给出来会发生什么。

例子一：二维数组

import torchx = torch.randn(2, 4)
print(x)
'''
tensor([[ 1.2864, -0.5955,  1.5042,  0.5398],[-1.2048,  0.5106, -2.0288,  1.4782]])
'''# y0表示矩阵dim=0维度上（每一列）张量最大值的索引
y0 = torch.argmax(x, dim=0)
print(y0)
'''
tensor([0, 1, 0, 1])
'''# y1表示矩阵dim=1维度上（每一行）张量最大值的索引
y1 = torch.argmax(x, dim=1)
print(y1)
'''
tensor([2, 3])
'''

例子二：三维数组

x = torch.randn(2, 4, 5)
print(x)
'''
tensor([[[-1.2204, -0.6428, -0.2278,  0.5589,  1.1589],[ 0.4235,  1.9663,  0.5055, -1.3472,  1.3523],[ 1.4220,  0.7886, -1.0821,  0.6268, -0.9465],[-0.3950,  1.3275,  0.3369,  1.0224, -0.9944]],[[ 0.6024, -0.2604, -0.8631,  0.8113, -0.3140],[ 0.3487, -0.1941, -0.3955, -0.1719, -1.3734],[ 0.2467, -0.4268, -1.3428,  0.7346,  1.0932],[-0.5799,  0.0976, -1.9403, -0.2643,  0.7657]]])
'''# dim=0，将第一个维度消除，也就是将两个[4*5]矩阵只保留一个，因此要在上下两个[3*4]的矩阵分别在对应位置上比较
y0 = torch.argmax(x, dim=0)
print(y0)
'''
tensor([[1, 1, 0, 1, 0],[0, 0, 0, 1, 0],[0, 0, 0, 1, 1],[0, 0, 0, 0, 1]])
'''# dim=1，将第二个维度消除，也就是将四个[2*5]矩阵只保留一个
y1 = torch.argmax(x, dim=1)
print(y1)
'''
tensor([[2, 1, 1, 3, 1],[0, 3, 1, 0, 2]])
'''y2 = torch.argmax(x, dim=2)
print(y2)
'''
tensor([[4, 1, 0, 1],[3, 0, 4, 4]])
'''

torch.argmax()函数相关推荐

【Pytorch】torch.argmax 函数详解
文章目录一.一个参数时的 torch.argmax 函数 1. 介绍 2. 实例二.多个参数时的 torch.argmax 函数 1. 介绍 2. 实例实例1:二维矩阵实例2:三维矩阵实例3 ...
Pytorch中torch.argmax()函数解析
一. torch.argmax()函数解析 1. 官网链接 torch.argmax(),如下图所示: 2. torch.argmax(input)函数解析 torch.argmax(input) → ...
torch.argmax函数说明
torch.argmax()函数 argmax函数:torch.argmax(input, dim=None, keepdim=False)返回指定维度最大值的序号,dim给定的定义是:the dem ...
【torch.argmax与torch.max详解】
Pytorch常用函数一.torch.max 1.调用方式 2.相关介绍 3.代码实例及图示理解二.torch.argmax 1.调用方式 2.相关介绍 3.代码实例及图示理解三.torch.m ...
numpy使用np.argmax函数获取一维数组中最大值所在的索引(index of largest value in numpy array with np.argmax)
numpy使用np.argmax函数获取一维数组中最大值所在的索引(index of largest value in numpy array with np.argmax) 目录 numpy使用np ...
pandas使用argmax函数返回给定series对象中最大值（max、maximum）的行索引实战
pandas使用argmax函数返回给定series对象中最大值(max.maximum)的行索引实战目录 pandas使用argmax函数返回给定series对象中最大值的行索引实战
numpy之argmax()函数
语法格式: numpy.argmax(a,axis) 作用:返回axis轴方向最大值的索引 a :为所需处理的矩阵 axis :为处理的轴向,axis=1为横轴方向,方向从左到右:axis=0为纵轴方 ...
代码阅读 | torch.sort函数
知识点https://blog.csdn.net/u012495579/article/details/106117511网友讲得非常清晰了. 代码如下: lengths = [17,17,19,23 ...
gather torch_浅谈Pytorch中的torch.gather函数的含义
pytorch中的gather函数 pytorch比tensorflow更加编程友好,所以准备用pytorch试着做最近要做的一些实验. 立个flag开始学习pytorch,新开一个分类整理学习pyt ...

torch.argmax()函数

torch.argmax()函数相关推荐

最新文章

热门文章