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算法之栈(stack)的实现
Aug 18 Python
Python描述器descriptor详解
Feb 03 Python
Python中的自省(反射)详解
Jun 02 Python
Python3实现发送QQ邮件功能(文本)
Dec 15 Python
Python实现拷贝/删除文件夹的方法详解
Aug 29 Python
pyqt5利用pyqtDesigner实现登录界面
Mar 28 Python
Python更新所有已安装包的操作
Feb 13 Python
Python基于pyjnius库实现访问java类
Jul 31 Python
Python 删除List元素的三种方法remove、pop、del
Nov 16 Python
利用python绘制正态分布曲线
Jan 04 Python
我对PyTorch dataloader里的shuffle=True的理解
May 20 Python
Python还能这么玩之用Python做个小游戏的外挂
Jun 04 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
phpmyadmin的#1251问题
2006/11/25 PHP
浅谈php优化需要注意的地方
2014/11/27 PHP
PHP模糊查询的实现方法(推荐)
2016/09/06 PHP
php微信公众号开发模式详解
2016/11/28 PHP
js类中获取外部函数名的方法与代码
2007/09/12 Javascript
javascript js cookie的存储,获取和删除
2007/12/29 Javascript
JQuery 将元素显示在屏幕的中央的代码
2010/02/27 Javascript
输入框的字数时时统计—关于 onpropertychange 和 oninput 使用
2011/10/21 Javascript
js兼容的placeholder属性详解
2013/08/18 Javascript
简单实用的全选反选按钮例子
2013/10/18 Javascript
jQuery浏览器CSS3特写兼容实例
2015/01/19 Javascript
JavaScript中操作字符串之localeCompare()方法的使用
2015/06/06 Javascript
jQuery基础知识点总结(必看)
2016/05/31 Javascript
jQuery实现的tab标签切换效果示例
2016/09/05 Javascript
BootStrap table删除指定行的注意事项(笔记整理)
2017/02/05 Javascript
vue.js中Vue-router 2.0基础实践教程
2017/05/08 Javascript
jQuery+ajax实现修改密码验证功能实例详解
2017/07/06 jQuery
Vue组件中prop属性使用说明实例代码详解
2018/05/31 Javascript
javascript中this的用法实践分析
2019/07/29 Javascript
离线安装Pyecharts的步骤以及依赖包流程
2020/04/23 Python
Python金融数据可视化汇总
2017/11/17 Python
Python安装whl文件过程图解
2020/02/18 Python
python实现FTP循环上传文件
2020/03/20 Python
css3个性化字体_动力节点Java学院整理
2017/07/12 HTML / CSS
css3 按钮样式简单可扩展创建
2013/03/18 HTML / CSS
iphoneX 适配客户端H5页面的方法教程
2017/12/08 HTML / CSS
英国精品买手店:Browns Fashion
2016/09/29 全球购物
日本最大的旅游网站:Rakuten Travel(乐天旅游)
2018/08/02 全球购物
C语言中一个结构不能包含指向自己的指针吗
2012/05/25 面试题
自我评价范文分享
2014/01/04 职场文书
违反课堂纪律检讨书
2014/01/19 职场文书
美国旅游签证工作证明
2014/10/14 职场文书
秋菊打官司观后感
2015/06/03 职场文书
Vue接口封装的完整步骤记录
2021/05/14 Vue.js
Python 详解通过Scrapy框架实现爬取CSDN全站热榜标题热词流程
2021/11/11 Python
人工智能深度学习OpenAI baselines的使用方法
2022/05/20 Python