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标准库内置函数complex介绍
Nov 25 Python
Win10下Python环境搭建与配置教程
Nov 18 Python
详解python里使用正则表达式的全匹配功能
Oct 19 Python
python中Apriori算法实现讲解
Dec 10 Python
Python设计模式之命令模式简单示例
Jan 10 Python
python matplotlib库绘制散点图例题解析
Aug 10 Python
python 实现让字典的value 成为列表
Dec 16 Python
Python 脚本的三种执行方式小结
Dec 21 Python
Pytorch实现各种2d卷积示例
Dec 30 Python
python实现批量转换图片为黑白
Jun 16 Python
Django中使用Celery的方法步骤
Dec 07 Python
matplotlib 范围选区(SpanSelector)的使用
Feb 24 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
无法载入 mcrypt 扩展,请检查 PHP 配置终极解决方案
2011/07/18 PHP
php代码审计比较有意思的例子
2014/05/07 PHP
php实现SAE上使用storage上传与下载文件的方法
2015/06/29 PHP
php常量详细解析
2015/10/27 PHP
Laravel+jQuery实现AJAX分页效果
2016/09/14 PHP
如何使用php等比例缩放图片
2016/10/12 PHP
JavaScript 闭包深入理解(closure)
2009/05/27 Javascript
Javascript 事件流和事件绑定
2009/07/16 Javascript
JavaScript浏览器选项卡效果
2010/08/25 Javascript
JS构造函数与原型prototype的区别介绍
2016/07/04 Javascript
JavaScript 事件对内存和性能的影响
2017/01/22 Javascript
layui分页效果实现代码
2017/05/19 Javascript
jQuery插件jsonview展示json数据
2018/05/26 jQuery
详解vue-loader在项目中是如何配置的
2018/06/04 Javascript
微信小程序中使用echarts的实现方法
2019/04/24 Javascript
[02:38]DOTA2亚洲邀请赛小组赛精彩集锦:Wings完美团击溃对手
2017/03/29 DOTA
python嵌套函数使用外部函数变量的方法(Python2和Python3)
2016/01/31 Python
详解python单元测试框架unittest
2018/07/02 Python
python3基于TCP实现CS架构文件传输
2018/07/28 Python
使用11行Python代码盗取了室友的U盘内容
2018/10/23 Python
python批量修改文件夹及其子文件夹下的文件内容
2019/03/15 Python
使用PyInstaller将Pygame库编写的小游戏程序打包为exe文件及出现问题解决方法
2019/09/06 Python
Django实现文件上传下载
2019/10/06 Python
python绘制动态曲线教程
2020/02/24 Python
详解css3中 text-fill-color属性
2019/07/08 HTML / CSS
BSTN意大利:德国街头和运动文化高品质商店
2020/12/22 全球购物
网上开商店的创业计划书
2014/01/19 职场文书
护士自我评价范文
2014/01/25 职场文书
家长建议怎么写
2014/05/15 职场文书
2015年幼儿园学期工作总结
2015/05/22 职场文书
男方家长婚礼答谢词
2015/09/29 职场文书
分析Python感知线程状态的解决方案之Event与信号量
2021/06/16 Python
一篇文章带你了解Python和Java的正则表达式对比
2021/09/15 Python
警用民用对讲机找不同
2022/02/18 无线电
聊聊基于pytorch实现Resnet对本地数据集的训练问题
2022/03/25 Python
html中两种获取标签内的值的方法
2022/06/10 HTML / CSS