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函数用法详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Django中自定义admin---Xadmin的实现代码，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值 ,需要的朋友可以参考下2019-08-08
这篇文章主要介绍了Python使用wget实现下载网络文件功能,简单介绍了wget安装以及Python使用wget下载tar格式网络文件并进行解压处理相关操作技巧,需要的朋友可以参考下2018-05-05
这篇文章主要介绍了python-地图可视化组件folium的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-12-12
下面小编就为大家带来一篇python 递归遍历文件夹,并打印满足条件的文件路径实例。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-08-08
不管你承不承认，数据清洗着实不是一件简单的任务，大多数情况下这项工作是十分耗时而乏味的，但它又是十分重要的，本篇文章带给你八个实例代码2022-01-01
os.path模块主要用于文件的属性获取，在编程中经常用到，本文将带你熟悉这个模块并掌握它的用法，感兴趣的朋友跟小编来看看吧2021-10-10
python使用paramiko执行服务器脚本并拿到实时结果
这篇文章主要介绍了python使用paramiko执行服务器脚本并拿到实时结果，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-12-12
这篇文章主要介绍了Python字典操作详细介绍及字典内建方法分享，具有一定借鉴价值,需要的朋友可以参考下2018-01-01
Socket是网络编程的一个抽象概念,通常我们用一个Socket表示“打开了一个网络链接”,而打开一个Socket需要知道目标计算机的IP地址和端口号,再指定协议类型即可,这篇文章主要给大家介绍了关于Python Socket编程的相关资料,需要的朋友可以参考下2021-08-08
Python BeautifulSoup基本用法详解(通过标签及class定位元素)
这篇文章主要介绍了Python BeautifulSoup基本用法(通过标签及class定位元素),本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-08-08
最新评论