torch.max()及其参数探究/torch.max()返回的索引是什么意思

注：根据1.7.1文档， torch.Tensor.max与torch.max用法相同

torch.max(input) → Tensor

返回input中最大的元素

实例：

 torch.max(input, dim, keepdim=False, *, out=None) -> (Tensor, LongTensor)

Parameters

    input (Tensor) – 输入张量dim (int) – 选择reduce的维keepdim (bool) – 上述的维度是否保留。 Default: False.

Keyword Arguments

out (tuple, optional) – 返回最大值和索引

需要注意的是，如果有多个最大值，返回第一个索引

实例：

从这个例子中，我们可以看出torch.Tensor.max与torch.max的结果是完全一致的。
再详细分析一下这个例子。
dim参数就是选择指定的维度，在该维度上的所有数据进行max查找，x.max(1)同理。
关于其返回的索引，我们可以这样理解。

torch.return_types.max(
values=tensor([6, 8, 5]),
indices=tensor([1, 2, 2]))

自然有：
indices[0]=1
indices[1]=2
indices[2]=3

由于我们选择的dim为0，在二维中，也就是“行”
所以，我们可以这样使用索引
将第dim维，换成indices[*]，另一维使用indices自己的索引。
（我们选择的是第0维，也就是说我们找出的是每一列的最大值。）

同时，我们扩充到3x3x3的数据试试

显然，同理，搜索每一组数据的最大数

索引方面，由于是一个二维数组

均同理

torch.max()及其参数探究/torch.max()返回的索引是什么意思相关推荐

Python3之max key参数学习记录
今天用Python写脚本,想要实现这样的功能:对于给定的字典,返回其中Value最大值对应的Key. 搜索后找到了解决方法,同时也学到了max key参数的作用. 例1, testlist = [9. ...
[转]Python3之max key参数学习记录
Python3之max key参数学习记录转自https://www.cnblogs.com/zhangwei22/p/9892422.html 今天用Python写脚本,想要实现这样的功能:对于给 ...
学习笔记5-知识点【卷积计算，dim，超参数，torch.zeros，gamma，beta，eps，assert，优化器，groups，// 和/，reshape，isinstance，hasattr
1.python 运行程序出现:"Caught IndexError in DataLoader worker process 0" 解决方法:可能是多进程的原因.查看代码,找到超 ...
dfs.datanode.max.xcievers参数导致hbase集群报错
2013/08/09 转发自http://bkeep.blog.163.com/blog/static/123414290201272644422987/ [案例]dfs.datanode.max.x ...
【无标题】Kafka副本管理—— 为何去掉replica.lag.max.messages参数
1.概述转载:Kafka副本管理-- 为何去掉replica.lag.max.messages参数今天查看Kafka 0.10.0的官方文档,发现了这样一句话:Configuration para ...
习题 9.5 建立一个对象数组，内放5个学生的数据（学号、成绩），设立一个函数max，用指向对象的指针作函数参数，在max函数中找出5个学生中成绩最高者，并输出其学号。
C++程序设计(第三版) 谭浩强习题9.5 个人设计习题 9.5 建立一个对象数组,内放5个学生的数据(学号.成绩),设立一个函数max,用指向对象的指针作函数参数,在max函数中找出5个学生中成 ...
Pytorch 类型错误：Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor.
Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor Pytorcht调试过程中,将数据传入模 ...
Python基础第六天——函数的使用、函数的参数、函数的返回值、函数的嵌套、命名空间、作用域、函数对象...
一.拾遗 1.通常情况下无参函数只是进行操作,所以不需要返回值:有参数函数则需要返回值. 2.练习: 判断用户输入一串字符,如果这串字符的长度小于5则重新输入,如果大于5则退出循环. while Tr ...
java新特性-函数式接口-作为方法参数-作为方法的返回值-常用函数式接口-Supplier-Consumer-Predicate-Function
文章目录函数式接口概念函数式接口作为方法参数函数式接口作为方法的返回值常用函数式接口 Supplier接口常用函数式接口 Consumer 函数式接口之 Predicate接口常用接口之 ...
Matlab函数语法(函数可以接受多个输入参数，并可能返回多个输出参数)
一个函数是一组在一起执行任务的语句. 在MATLAB中,函数在单独的文件中定义.文件的名称和函数的名称应该是一样的. 函数在自己的工作空间内的变量上运行,这个变量也称为本地工作空间,与在MATLAB命 ...

torch.max()及其参数探究/torch.max()返回的索引是什么意思

torch.max()及其参数探究/torch.max()返回的索引是什么意思相关推荐

最新文章

热门文章