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 08 Python
Django框架中数据的连锁查询和限制返回数据的方法
Jul 17 Python
Python脚本实时处理log文件的方法
Nov 21 Python
Python之时间和日期使用小结
Feb 14 Python
python3.6下Numpy库下载与安装图文教程
Apr 02 Python
Python检测数据类型的方法总结
May 20 Python
浅谈Python3实现两个矩形的交并比(IoU)
Jan 18 Python
python 实现在无序数组中找到中位数方法
Mar 03 Python
keras:model.compile损失函数的用法
Jul 01 Python
Python如何爬取51cto数据并存入MySQL
Aug 25 Python
Python系统公网私网流量监控实现流程
Nov 23 Python
python drf各类组件的用法和作用
Jan 12 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/10/30 PHP
javascript的事件描述
2006/09/08 Javascript
Js 获取当前日期时间及其它操作实现代码
2021/03/04 Javascript
JS DOM 操作实现代码
2010/08/01 Javascript
javascript面向对象入门基础详细介绍
2012/09/05 Javascript
javascript 模拟坦克大战游戏(html5版)附源码下载
2014/04/08 Javascript
jquery实现在页面加载的时自动为日期插件添加当前日期
2014/08/20 Javascript
JCrop+ajaxUpload 图像切割上传的实例代码
2016/07/20 Javascript
jQuery元素属性操作实例(设置、获取及删除元素属性)
2016/09/08 Javascript
jQuery插件echarts去掉垂直网格线用法示例
2017/03/03 Javascript
JavaScript中闭包的详解
2017/04/01 Javascript
在JS中如何把毫秒转换成规定的日期时间格式实例
2017/05/11 Javascript
Nginx 配置多站点vhost 的方法
2018/01/07 Javascript
Vue中render函数的使用方法
2018/01/31 Javascript
原生JS实现瀑布流插件
2018/02/06 Javascript
Bootstrap Paginator+PageHelper实现分页效果
2018/12/29 Javascript
微信小程序wxml列表渲染原理解析
2019/11/27 Javascript
vue+iview实现分页及查询功能
2020/11/17 Vue.js
在vue中动态修改css其中一个属性值操作
2020/12/07 Vue.js
Python设计模式之工厂模式简单示例
2018/01/09 Python
python实现弹跳小球
2019/05/13 Python
Django中的cookie和session
2019/08/27 Python
python字符串下标与切片及使用方法
2020/02/13 Python
python中有函数重载吗
2020/05/28 Python
x-ua-compatible content=”IE=7, IE=9″意思理解
2013/07/22 HTML / CSS
YOOX美国官方网站:全球著名的多品牌时尚网络概念店
2016/09/11 全球购物
关于幼儿的自我评价
2013/12/18 职场文书
高中体育教学反思
2014/01/24 职场文书
2014年驾驶员工作总结
2014/11/18 职场文书
2015年党支部公开承诺书
2015/01/22 职场文书
大学学生会竞选稿
2015/11/19 职场文书
接收函
2019/04/22 职场文书
浅谈Python基础之列表那些事儿
2021/05/11 Python
redis实现共同好友的思路详解
2021/05/26 Redis
python ansible自动化运维工具执行流程
2021/06/24 Python
豆瓣2021评分最高动画剧集-豆瓣评分最高的动画剧集2021
2022/03/18 日漫