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 相关文章推荐
web.py中调用文件夹内模板的方法
Aug 26 Python
Python写入数据到MP3文件中的方法
Jul 10 Python
Python实现短网址ShortUrl的Hash运算实例讲解
Aug 10 Python
Python对数据库操作
Mar 28 Python
Python网络爬虫实例讲解
Apr 28 Python
Python实现将一个大文件按段落分隔为多个小文件的简单操作方法
Apr 17 Python
python3实现名片管理系统
Nov 29 Python
python画图的函数用法以及技巧
Jun 28 Python
FFrpc python客户端lib使用解析
Aug 24 Python
解决Pycharm 包已经下载,但是运行代码提示找不到模块的问题
Aug 31 Python
在python3中实现更新界面
Feb 21 Python
python爬虫把url链接编码成gbk2312格式过程解析
Jun 08 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中如何在有限的内存中读取大文件
2013/07/02 PHP
php缩小png图片不损失透明色的解决方法
2013/12/25 PHP
php无限级分类实现方法分析
2016/10/19 PHP
javascript  Error 对象 错误处理
2008/05/18 Javascript
jQuery点击后一组图片左右滑动的实现代码
2012/08/16 Javascript
你必须知道的Javascript知识点之"单线程事件驱动"的使用
2013/04/23 Javascript
javascript折半查找详解
2015/01/26 Javascript
深入浅析JavaScript面向对象和原型函数
2016/02/06 Javascript
jQuery ajax提交Form表单实例(附demo源码)
2016/04/06 Javascript
jquery实现全选功能效果的实现代码
2016/05/05 Javascript
js实现四舍五入完全保留两位小数的方法
2016/08/02 Javascript
浅谈js的异步执行
2016/10/18 Javascript
Bootstrap基本插件学习笔记之标签切换(17)
2016/12/08 Javascript
Node.js对MongoDB数据库实现模糊查询的方法
2017/05/03 Javascript
vue2单元测试环境搭建
2018/05/24 Javascript
js获取本日、本周、本月的时间代码
2020/02/01 Javascript
[52:44]VGJ.T vs infamous Supermajor小组赛D组败者组第一轮 BO3 第一场 6.3
2018/06/04 DOTA
python 获取et和excel的版本号
2009/04/09 Python
浅谈python jieba分词模块的基本用法
2017/11/09 Python
Python2.X/Python3.X中urllib库区别讲解
2017/12/19 Python
Python实现二维数组输出为图片
2018/04/03 Python
如何用Python合并lmdb文件
2018/07/02 Python
django用户登录验证的完整示例代码
2019/07/21 Python
pytorch中tensor.expand()和tensor.expand_as()函数详解
2019/12/27 Python
Python远程linux执行命令实现
2020/11/11 Python
HTML5+JS实现俄罗斯方块原理及具体步骤
2013/11/29 HTML / CSS
大学毕业感言100字
2014/02/03 职场文书
优秀毕业生推荐信范文
2014/03/07 职场文书
我的大学生活演讲稿
2014/04/25 职场文书
做一个有道德的人活动方案
2014/08/25 职场文书
投标承诺函格式
2015/01/21 职场文书
《实心球》教学反思
2016/02/23 职场文书
煤矿安全生产管理协议书
2016/03/22 职场文书
使用这 6个Vue加载动画库来减少我们网站的跳出率
2021/05/18 Vue.js
Mysql8.0递归查询的简单用法示例
2021/08/04 MySQL
opencv深入浅出了解机器学习和深度学习
2022/03/17 Python