pytorch中的 .view()函数的用法介绍


Posted in Python onMarch 17, 2022

一、普通用法 (手动调整size)

view()相当于reshape、resize,重新调整Tensor的形状。

import torch
a1 = torch.arange(0,16)
print(a1)
# tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])
a2 = a1.view(8, 2)
a3 = a1.view(2, 8)
a4 = a1.view(4, 4)
print(a2)
#tensor([[ 0,  1],
#        [ 2,  3],
#        [ 4,  5],
#        [ 6,  7],
#        [ 8,  9],
#        [10, 11],
#        [12, 13],
#        [14, 15]])
print(a3)
#tensor([[ 0,  1,  2,  3,  4,  5,  6,  7],
#        [ 8,  9, 10, 11, 12, 13, 14, 15]])
print(a4)
#tensor([[ 0,  1,  2,  3],
#        [ 4,  5,  6,  7],
#        [ 8,  9, 10, 11],
#        [12, 13, 14, 15]])

二、特殊用法:参数-1 (自动调整size)

view中一个参数定为-1,代表自动调整这个维度上的元素个数,以保证元素的总数不变。

v1 = torch.arange(0,16)
print(v1)
# tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])
v2 = v1.view(-1, 16)
v2
# tensor([[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15]])
v2 = v1.view(-1, 8)
v2
# tensor([[ 0,  1,  2,  3,  4,  5,  6,  7],
#         [ 8,  9, 10, 11, 12, 13, 14, 15]])
v2 = v1.view(-1, 4)
v2
#tensor([[ 0,  1,  2,  3],
#        [ 4,  5,  6,  7],
#        [ 8,  9, 10, 11],
#        [12, 13, 14, 15]])
v2 = v1.view(-1, 2)
v2
#tensor([[ 0,  1],
#        [ 2,  3],
#        [ 4,  5],
#        [ 6,  7],
#        [ 8,  9],
#        [10, 11],
#        [12, 13],
#        [14, 15]])
v3 = v1.view(4*4, -1)
v3
# tensor([[ 0],
#         [ 1],
#         [ 2],
#         [ 3],
#         [ 4],
#         [ 5],
#         [ 6],
#         [ 7],
#         [ 8],
#         [ 9],
#         [10],
#         [11],
#         [12],
#         [13],
#         [14],
#         [15]])

到此这篇关于pytorch中的 .view()函数的用法介绍的文章就介绍到这了,更多相关pytorch .view()函数内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
用Python的urllib库提交WEB表单
Feb 24 Python
python 域名分析工具实现代码
Jul 15 Python
Python中join和split用法实例
Apr 14 Python
python PIL模块与随机生成中文验证码
Feb 27 Python
tensorflow学习教程之文本分类详析
Aug 07 Python
Python面向对象之Web静态服务器
Sep 03 Python
Django框架 查询Extra功能实现解析
Sep 04 Python
TensorFlow实现指数衰减学习率的方法
Feb 05 Python
Python无损压缩图片的示例代码
Aug 06 Python
python自动打开浏览器下载zip并提取内容写入excel
Jan 04 Python
教你怎么用Python生成九宫格照片
May 20 Python
图文详解matlab原始处理图像几何变换
Jul 09 Python
Python绘画好看的星空图
opencv深入浅出了解机器学习和深度学习
Python与C++中梯度方向直方图的实现
JAVA SpringMVC实现自定义拦截器
Mar 16 #Python
Python Pandas 删除列操作
Mar 16 #Python
Python实现批量自动整理文件
Mar 16 #Python
Pandas-DataFrame知识点汇总
Mar 16 #Python
You might like
用PHP控制用户的浏览器--ob*函数的使用说明
2007/03/16 PHP
php 执行系统命令的方法
2009/07/07 PHP
php 文章采集正则代码
2009/12/28 PHP
PHP is_subclass_of函数的一个BUG和解决方法
2014/06/01 PHP
PHP类的特性实例分析
2016/09/28 PHP
php自定义函数实现统计中文字符串长度的方法小结
2017/04/15 PHP
学习js所必须要知道的一些
2007/03/07 Javascript
JavaScript DOM 添加事件
2009/02/14 Javascript
Javascript拓展String方法小结
2013/07/08 Javascript
兼容FF和IE的动态table示例自写
2013/10/21 Javascript
JS不间断向上滚动效果代码
2013/12/25 Javascript
获取3个数组不重复的值的具体实现
2013/12/30 Javascript
JS获取当前日期时间并定时刷新示例
2021/03/04 Javascript
使用百度地图api实现根据地址查询经纬度
2014/12/11 Javascript
AngularJS实现一次监听多个值发生的变化
2016/08/31 Javascript
Bootstrap导航条鼠标悬停下拉菜单
2017/01/04 Javascript
微信小程序开发之相册选择和拍照详解及实例代码
2017/02/22 Javascript
node.js操作MongoDB的实例详解
2017/10/11 Javascript
快速解决vue在ios端下点击响应延时的问题
2018/08/27 Javascript
vuex实现像调用模板方法一样调用Mutations方法
2019/11/06 Javascript
Vue如何基于vue-i18n实现多国语言兼容
2020/07/17 Javascript
Python中为feedparser设置超时时间避免堵塞
2014/09/28 Python
Python获取Linux系统下的本机IP地址代码分享
2014/11/07 Python
python实现多线程的方式及多条命令并发执行
2016/06/07 Python
Python输出带颜色的字符串实例
2017/10/10 Python
python下10个简单实例代码
2017/11/15 Python
python 的 openpyxl模块 读取 Excel文件的方法
2019/09/09 Python
python装饰器相当于函数的调用方式
2019/12/27 Python
python实现加密的方式总结
2020/01/19 Python
Python3 pickle对象串行化代码实例解析
2020/03/23 Python
Tensorflow中批量读取数据的案列分析及TFRecord文件的打包与读取
2020/06/30 Python
Python WebSocket长连接心跳与短连接的示例
2020/11/24 Python
趣味运动会活动方案
2014/02/12 职场文书
2015年四年级班主任工作总结
2015/10/22 职场文书
委托书范本格式
2019/04/18 职场文书
2019教师的学习计划
2019/06/25 职场文书