pytorch中torch.max和Tensor.view函数用法详解
更新时间：2020年01月03日 15:07:28 作者：坎幽黑尔弥？
今天小编就为大家分享一篇pytorch中torch.max和Tensor.view函数用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
torch.max()
1.
torch.max()简单来说是返回一个tensor中的最大值。
例如：
>>> si=torch.randn(4,5)
>>> print(si)
tensor([[ 1.1659, -1.5195, 0.0455, 1.7610, -0.2064],
[-0.3443, 2.0483, 0.6303, 0.9475, 0.4364],
[-1.5268, -1.0833, 1.6847, 0.0145, -0.2088],
[-0.8681, 0.1516, -0.7764, 0.8244, -1.2194]])
>>> print(torch.max(si))
tensor(2.0483)
2.
这个函数的参数中还有一个dim参数，使用方法为re = torch.max(Tensor,dim),返回的re为一个二维向量，其中re[0]为最大值的Tensor，re[1]为最大值对应的index的Tensor。
例如：
>>> print(torch.max(si,0)[0]) tensor([1.1659, 2.0483, 1.6847, 1.7610, 0.4364])
注意，Tensor的维度从0开始算起。在torch.max()中指定了dim之后，比如对于一个3x4x5的Tensor，指定dim为0后，得到的结果是维度为0的“每一行”对应位置求最大的那个值，此时输出的Tensor的维度是4x5.
对于简单的二维Tensor，如上面例子的这个4x5的Tensor。指定dim为0，则给出的结果是4行做比较之后的最大值；如果指定dim为1，则给出的结果是5列做比较之后的最大值，且此处做比较时是按照位置分别做比较，得到一个新的Tensor。
Tensor.view()
简单说就是一个把tensor 进行reshape的操作。
>>> a=torch.randn(3,4,5,7) >>> b = a.view(1,-1) >>> print(b.size()) torch.Size([1, 420])
其中参数-1表示剩下的值的个数一起构成一个维度。如上例中，第一个参数1将第一个维度的大小设定成1，后一个-1就是说第二个维度的大小=元素总数目/第一个维度的大小，此例中为3*4*5*7/1=420.
>>> d = a.view(a.size(0),a.size(1),-1) >>> print(d.size()) torch.Size([3, 4, 35]) >>> e=a.view(4,-1,5) >>> print(e.size()) torch.Size([4, 21, 5])
以上这篇pytorch中torch.max和Tensor.view函数用法详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python控制台中实现进度条功能的方法，想要了解的朋友可以参考一下2015-11-11
这篇文章主要介绍了Python发送邮件功能,结合实例形式分析了Python使用QQ邮箱进行邮件发送的相关设置与使用技巧,需要的朋友可以参考下2018-12-12
这篇文章主要介绍了详解Python中的三器一闭,Python中的三器一闭是指迭代器、装饰器、生成器和闭包,需要的朋友可以参考下2023-05-05
在本篇文章里小编给大家整理了关于Python3中exp()函数用法以及相关知识点，需要的朋友们学习下。2019-02-02
数据分析与可视化是当今数据分析的发展方向,大数据时代,数据资源具有海量特征,数据分析和可视化主要通过Python数据分析来实现,本文给大家介绍了如何基于Python实现的影视数据智能分析系统,文中给出了部分详细代码,感兴趣的朋友跟着小编一起来看看吧2024-01-01
这篇文章主要介绍了python的super()的作用和原理，super(), 在类的继承里面super()非常常用， 它解决了子类调用父类方法的一些问题， 父类多次被调用时只执行一次， 优化了执行逻辑，下面我们就来详细看一下2020-10-10
这篇文章主要介绍了Python实现字典(dict)的迭代操作,结合实例形式分析了Python针对字典键、值以及键值对等迭代操作实现技巧,需要的朋友可以参考下2018-06-06
Python class类@staticmethod及@classmethod区别浅析
这篇文章主要为大家介绍了Python class类@staticmethod及@classmethod区别浅析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-07-07
在本篇文章里小编给大家分享的是一篇关于虚拟机下载python是否需要联网的相关文章，有需要的朋友们可以参考下。2020-07-07
简单掌握Python的Collections模块中counter结构的用法
counter数据结构被用来提供技术功能,形式类似于Python中内置的字典结构,这里通过几个小例子来简单掌握Python的Collections模块中counter结构的用法:2016-07-07
最新评论