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之eval()函数危险性浅析
Jul 03 Python
Python中的random()方法的使用介绍
May 15 Python
python中__slots__用法实例
Jun 04 Python
Python判断列表是否已排序的各种方法及其性能分析
Jun 20 Python
Python 实现简单的shell sed替换功能(实例讲解)
Sep 29 Python
python+matplotlib绘制旋转椭圆实例代码
Jan 12 Python
Python Pexpect库的简单使用方法
Jan 29 Python
python3 实现调用串口功能
Dec 26 Python
python判断变量是否为int、字符串、列表、元组、字典的方法详解
Feb 13 Python
python对execl 处理操作代码
Jun 22 Python
解决Django响应JsonResponse返回json格式数据报错问题
Aug 09 Python
Pytorch模型迁移和迁移学习,导入部分模型参数的操作
Mar 03 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实现的随机广告显示代码
2007/06/14 PHP
php和javascript之间变量的传递实现代码
2012/12/19 PHP
linux实现php定时执行cron任务详解
2013/12/24 PHP
Yii2中简单的场景使用介绍
2017/06/02 PHP
javascript 写类方式之七
2009/07/05 Javascript
JS 巧妙获取剪贴板数据 Excel数据的粘贴
2009/07/09 Javascript
jQuery实现随意改变div任意属性的名称和值(部分原生js实现)
2013/05/28 Javascript
json格式的时间显示为正常年月日的方法
2013/09/08 Javascript
百度判断手机终端并自动跳转js代码及使用实例
2014/06/11 Javascript
jquery控制背景音乐开关与自动播放提示音的方法
2015/02/06 Javascript
JS中parseInt()和map()用法分析
2016/12/16 Javascript
JS实现简单的浮动碰撞效果示例
2017/12/28 Javascript
Javascript网页抢红包外挂实现分享
2018/01/11 Javascript
bootstrap 点击空白处popover弹出框隐藏实例
2018/01/24 Javascript
在vue中给列表中的奇数行添加class的实现方法
2018/09/05 Javascript
vue+echarts实现可拖动节点的折线图(支持拖动方向和上下限的设置)
2019/04/12 Javascript
基于Taro的微信小程序模板消息-获取formId功能模块封装实践
2019/07/15 Javascript
Vue+Bootstrap收藏(点赞)功能逻辑与具体实现
2020/10/22 Javascript
JS闭包原理及其使用场景解析
2020/12/03 Javascript
利用Python脚本在Nginx和uwsgi上部署MoinMoin的教程
2015/05/05 Python
Python生成随机密码的方法
2017/06/16 Python
利用Python正则表达式过滤敏感词的方法
2019/01/21 Python
python 抓包保存为pcap文件并解析的实例
2019/07/23 Python
python数据类型之间怎么转换技巧分享
2019/08/20 Python
Python中logging日志库实例详解
2020/02/19 Python
python正则表达式实例代码
2020/03/03 Python
Python用来做Web开发的优势有哪些
2020/08/05 Python
在Python3.74+PyCharm2020.1 x64中安装使用Kivy的详细教程
2020/08/07 Python
Python Web项目Cherrypy使用方法镜像
2020/11/05 Python
CSS3制作气泡对话框的实例教程
2016/05/10 HTML / CSS
Farfetch中文官网:奢侈品牌时尚购物平台
2020/03/15 全球购物
新学期家长寄语
2014/01/19 职场文书
计划生育标语
2014/06/23 职场文书
民族精神月活动总结
2014/08/28 职场文书
公司人事管理制度
2015/08/05 职场文书
Redis持久化与主从复制的实践
2021/04/27 Redis