torch.argmax方法详解

torch.argmax(x, dim)，其中x为张量，dim控制比较的维度，返回最大值的索引。
1.当dim=0时

import torch
x = torch.rand(2, 3,2)
print(x)
torch.argmax(x, dim=0)

当dim=0时，表示后两个维度进行比较，得到结果如下图：

比较过程为：输出结果的张量y的大小为去掉需比较维度dim后的大小，即3x2。然后依次确定这6个值，首先，对x[:,0,0]中的值进行比较，

取较大值的索引值输出结果的值，0.6718>0.6402,即y[0,0]=1；接着，对x[:,0,1]进行比较，

取较大值的索引值输出结果的值，即y[0,1]=0；以此类推，直到将所有的比较完成。

当dim为1，或2的情况也类似，结果如下：
2.当dim=1

3.当dim=2

torch.argmax方法详解相关推荐

【Pytorch】torch.argmax 函数详解
文章目录一.一个参数时的 torch.argmax 函数 1. 介绍 2. 实例二.多个参数时的 torch.argmax 函数 1. 介绍 2. 实例实例1:二维矩阵实例2:三维矩阵实例3 ...
PyTorch中的torch.nn.Parameter() 详解
PyTorch中的torch.nn.Parameter() 详解今天来聊一下PyTorch中的torch.nn.Parameter()这个函数,笔者第一次见的时候也是大概能理解函数的用途,但是具体实 ...
python中backward_pytorch的梯度计算以及backward方法详解
基础知识 tensors: tensor在pytorch里面是一个n维数组.我们可以通过指定参数reuqires_grad=True来建立一个反向传播图,从而能够计算梯度.在pytorch中一般叫做d ...
pytorch模型(.pt)转onnx模型(.onnx)的方法详解(1)
1. pytorch模型转换到onnx模型 2.运行onnx模型 3.比对onnx模型和pytorch模型的输出结果我这里重点是第一点和第二点,第三部分比较容易首先你要安装依赖库:onnx ...
einsum方法详解（爱因斯坦求和）
einsum方法详解(爱因斯坦求和) einsum是pytorch.numpy中一个十分优雅的方法,如果利用得当,可完全代替所有其他的矩阵计算方法,不过这需要一定的学习成本.本文旨在详细解读einsu ...
python统计csv行数_对Python 多线程统计所有csv文件的行数方法详解
如下所示: #统计某文件夹下的所有csv文件的行数(多线程) import threading import csv import os class MyThreadLine(threading.Th ...
python修改文件内容_Python批量修改文本文件内容的方法详解
这篇文章主要介绍了Python批量修改文本文件内容的方法的相关资料,需要的朋友可以参考下 Python批量替换文件内容,支持嵌套文件夹 import os path="./" fo ...
python二维元组_python中读入二维csv格式的表格方法详解(以元组/列表形式表示)
如何去读取一个没有表头的二维csv文件(如下图所示)? 并以元组的形式表现数据: ((1.0, 0.0, 3.0, 180.0), (2.0, 0.0, 2.0, 180.0), (3.0, 0.0, ...
Spring JdbcTemplate方法详解
2019独角兽企业重金招聘Python工程师标准>>> Spring JdbcTemplate方法详解标签: springhsqldbjava存储数据库相关sql 2012-07- ...

torch.argmax方法详解

torch.argmax方法详解

torch.argmax方法详解相关推荐

最新文章

热门文章