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实现html转ubb代码(html2ubb)
Jul 03 Python
跟老齐学Python之私有函数和专有方法
Oct 24 Python
跟老齐学Python之网站的结构
Oct 24 Python
python中函数传参详解
Jul 03 Python
Django migrations 默认目录修改的方法教程
Sep 28 Python
python numpy数组的索引和切片的操作方法
Oct 20 Python
pandas dataframe的合并实现(append, merge, concat)
Jun 24 Python
python pycharm的安装及其使用
Oct 11 Python
Python+Redis实现布隆过滤器
Dec 08 Python
TensorFlow实现保存训练模型为pd文件并恢复
Feb 06 Python
python 安装移动复制第三方库操作
Jul 13 Python
详解Python调用系统命令的六种方法
Jan 28 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绝对路径与相对路径之间关系的的分析
2010/03/03 PHP
Fatal error: session_start(): Failed to initialize storage module: files问题解决方法
2014/05/04 PHP
PHP中使用GD库创建圆形饼图的例子
2014/11/19 PHP
php实现面包屑导航例子分享
2015/12/19 PHP
php制作基于xml的RSS订阅源功能示例
2017/02/08 PHP
JQuery Ajax通过Handler访问外部XML数据的代码
2010/06/01 Javascript
JavaScript中的undefined学习总结
2013/11/30 Javascript
JS中FRAME的操作问题实例分析
2014/10/21 Javascript
javascript比较两个日期相差天数的方法
2015/07/24 Javascript
jQuery带时间的日期控件代码分享
2015/08/26 Javascript
基于jQuery实现仿51job城市选择功能实例代码
2016/03/02 Javascript
javascript高级选择器querySelector和querySelectorAll全面解析
2016/04/07 Javascript
fullpage.js全屏滚动插件使用实例
2016/09/06 Javascript
JavaScript实现的选择排序算法实例分析
2017/04/14 Javascript
JS正则表达式完美实现身份证校验功能
2017/10/18 Javascript
分享vue.js devtools遇到一系列问题
2017/10/24 Javascript
VUE 动态组件的应用案例分析
2019/12/02 Javascript
[01:28:44]DOTA2-DPC中国联赛定级赛 RNG vs iG BO3第一场 1月10日
2021/03/11 DOTA
[52:20]DOTA2-DPC中国联赛正赛 SAG vs XGBO3 第一场 3月5日
2021/03/11 DOTA
linux系统使用python获取cpu信息脚本分享
2014/01/15 Python
python实现文件名批量替换和内容替换
2014/03/20 Python
python爬虫入门教程--利用requests构建知乎API(三)
2017/05/25 Python
python分析作业提交情况
2017/11/22 Python
Python实现程序判断季节的代码示例
2019/01/28 Python
使用Python Pandas处理亿级数据的方法
2019/06/24 Python
Python编写一个验证码图片数据标注GUI程序附源码
2019/12/09 Python
TensorFLow 变量命名空间实例
2020/02/11 Python
Django限制API访问频率常用方法解析
2020/10/12 Python
计算机求职信
2013/12/01 职场文书
党员服务承诺书
2014/05/28 职场文书
电子商务专业求职信
2014/07/10 职场文书
学校关爱留守儿童活动方案
2014/08/27 职场文书
领导干部四风问题自我剖析材料
2014/09/25 职场文书
党员先进事迹材料
2014/12/19 职场文书
服务员岗位职责
2015/02/03 职场文书
2016公司年会通知范文
2015/04/25 职场文书