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面向对象编程的知识点总结
Feb 14 Python
python 打印直角三角形,等边三角形,菱形,正方形的代码
Nov 21 Python
对pandas进行数据预处理的实例讲解
Apr 20 Python
Pandas实现数据类型转换的一些小技巧汇总
May 07 Python
利用Python yagmail三行代码实现发送邮件
May 11 Python
python实现黑客字幕雨效果
Jun 21 Python
Python实现多级目录压缩与解压文件的方法
Sep 01 Python
Python图像处理之颜色的定义与使用分析
Jan 03 Python
Python3.6中Twisted模块安装的问题与解决
Apr 15 Python
MNIST数据集转化为二维图片的实现示例
Jan 10 Python
浅谈PyTorch中in-place operation的含义
Jun 27 Python
Matplotlib中rcParams使用方法
Jan 05 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修改时间格式的代码
2011/05/29 PHP
PHP反转字符串函数strrev()函数的用法
2012/02/04 PHP
C# WinForm中实现快捷键自定义设置实例
2015/01/23 PHP
PHP curl使用实例
2015/07/02 PHP
php的laravel框架快速集成微信登录的方法
2016/12/12 PHP
PHP vsprintf()函数格式化字符串操作原理解析
2020/07/14 PHP
javascript 二维数组的实现与应用
2010/03/16 Javascript
Web开发者必备的12款超赞jQuery插件
2010/12/03 Javascript
javascript 学习笔记(一)DOM基本操作
2011/04/08 Javascript
JS实现悬浮移动窗口(悬浮广告)的特效
2013/03/12 Javascript
nodejs实现黑名单中间件设计
2014/06/17 NodeJs
js读取json的两种常用方法示例介绍
2014/10/19 Javascript
jQuery层级选择器用法分析
2015/02/10 Javascript
AngularJS equal比较对象实例详解
2016/09/14 Javascript
JavaScript注册时密码强度校验代码
2017/06/30 Javascript
使用jquery的jsonp如何发起跨域请求及其原理详解
2017/08/17 jQuery
React Native中Mobx的使用方法详解
2018/12/04 Javascript
如何在JS文件中获取Vue组件
2020/09/16 Javascript
Python+Selenium+PIL+Tesseract自动识别验证码进行一键登录
2017/09/20 Python
python将字符串以utf-8格式保存在txt文件中的方法
2018/10/30 Python
Python3.5装饰器典型案例分析
2019/04/30 Python
Python解释器及PyCharm工具安装过程
2020/02/26 Python
Python Selenium 设置元素等待的三种方式
2020/03/18 Python
Python实现清理微信僵尸粉功能示例【基于itchat模块】
2020/05/29 Python
Python Merge函数原理及用法解析
2020/09/16 Python
python中的垃圾回收(GC)机制
2020/09/21 Python
女孩每月服装订阅盒:kidpik
2019/04/17 全球购物
实现strstr功能,即在父串中寻找子串首次出现的位置
2016/08/05 面试题
百度JavaScript笔试题
2015/01/15 面试题
现在输入n个数字,以逗号,分开;然后可选择升或者降序排序;按提交键就在另一页面显示按什么排序,结果为,提供reset
2012/11/09 面试题
报关简历自我评价怎么写
2013/09/19 职场文书
综合素质的自我鉴定
2013/10/07 职场文书
新闻专业推荐信范文
2013/11/20 职场文书
大学新生欢迎词
2014/01/10 职场文书
雨雪天气温馨提示
2015/07/15 职场文书
2016中学教师读书心得体会
2016/01/13 职场文书