pytorch中torch.max和Tensor.view函数用法详解


Posted in Python onJanuary 03, 2020

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 相关文章推荐
python正则表达式抓取成语网站
Nov 20 Python
python网络编程示例(客户端与服务端)
Apr 24 Python
python使用心得之获得github代码库列表
Jun 25 Python
Python实现的ini文件操作类分享
Nov 20 Python
在Django的通用视图中处理Context的方法
Jul 21 Python
Python3实现发送QQ邮件功能(html)
Dec 15 Python
Python 中字符串拼接的多种方法
Jul 30 Python
使用Python实现文字转语音并生成wav文件的例子
Aug 08 Python
Python线程障碍对象Barrier原理详解
Dec 02 Python
python3.8与pyinstaller冲突问题的快速解决方法
Jan 16 Python
Pytorch框架实现mnist手写库识别(与tensorflow对比)
Jul 20 Python
Python scrapy爬取起点中文网小说榜单
Jun 13 Python
pytorch逐元素比较tensor大小实例
Jan 03 #Python
pytorch 改变tensor尺寸的实现
Jan 03 #Python
Pytorch Tensor 输出为txt和mat格式方式
Jan 03 #Python
CentOS7下安装python3.6.8的教程详解
Jan 03 #Python
Python实现大数据收集至excel的思路详解
Jan 03 #Python
Python如何基于rsa模块实现非对称加密与解密
Jan 03 #Python
PyTorch的自适应池化Adaptive Pooling实例
Jan 03 #Python
You might like
实用函数2
2007/11/08 PHP
Pain 全世界最小最简单的PHP模板引擎 (普通版)
2011/10/23 PHP
PHP实现根据设备类型自动跳转相应页面的方法
2014/07/24 PHP
PHP中使用数组指针函数操作数组示例
2014/11/19 PHP
php实现redis数据库指定库号迁移的方法
2015/01/14 PHP
PHP实现自动识别原编码并对字符串进行编码转换的方法
2016/07/13 PHP
PHP面向对象程序设计方法实例详解
2016/12/24 PHP
php使用PDO获取结果集的方法
2017/02/16 PHP
php redis setnx分布式锁简单原理解析
2020/10/23 PHP
使用jQuery模板来展现json数据的代码
2010/10/22 Javascript
JavaScript之编码规范 推荐
2012/05/23 Javascript
js获取网页高度(详细整理)
2012/12/28 Javascript
javascript新建标签,判断键盘输入,以及判断焦点(示例代码)
2013/11/25 Javascript
js的Prototype属性解释及常用方法
2014/05/08 Javascript
javascript上下方向键控制表格行选中并高亮显示的方法
2015/02/13 Javascript
浅析C/C++,Java,PHP,JavaScript,Json数组、对象赋值时最后一个元素后面是否可以带逗号
2016/03/22 Javascript
jQuery EasyUI常用数据验证汇总
2016/09/18 Javascript
jQuery Validate 数组 全部验证问题
2017/01/12 Javascript
vue中axios处理http发送请求的示例(Post和get)
2017/10/13 Javascript
Vue数据驱动表单渲染,轻松搞定form表单
2019/07/19 Javascript
javascript实现动态时钟的启动和停止
2020/07/29 Javascript
手把手教您实现react异步加载高阶组件
2020/04/07 Javascript
[03:06]2018年度CS GO最具人气解说-完美盛典
2018/12/16 DOTA
python调用短信猫控件实现发短信功能实例
2014/07/04 Python
ubuntu17.4下为python和python3装上pip的方法
2018/06/12 Python
python3 对list中每个元素进行处理的方法
2018/06/29 Python
解决Mac下使用python的坑
2019/08/13 Python
python opencv如何实现图片绘制
2020/01/19 Python
Janie and Jack美国官网:GAP旗下的高档童装品牌
2019/09/09 全球购物
质检部职责
2013/12/28 职场文书
大学生通用个人的自我评价
2014/02/10 职场文书
大学学生会竞选演讲稿
2014/04/25 职场文书
党员个人批评与自我批评
2014/10/14 职场文书
公民授权委托书
2014/10/15 职场文书
公司开除员工通知
2015/04/22 职场文书
医疗纠纷调解协议书
2015/08/06 职场文书