pytorch 在sequential中使用view来reshape的例子


Posted in Python onAugust 20, 2019

pytorch中view是tensor方法,然而在sequential中包装的是nn.module的子类,

因此需要自己定义一个方法:

import torch.nn as nn
class Reshape(nn.Module):
 def __init__(self, *args):
  super(Reshape, self).__init__()
  self.shape = args

 def forward(self, x):
  # 如果数据集最后一个batch样本数量小于定义的batch_batch大小,会出现mismatch问题。可以自己修改下,如只传入后面的shape,然后通过x.szie(0),来输入。
  return x.view(self.shape)
class Reshape(nn.Module):
 def __init__(self, *args):
  super(Reshape, self).__init__()
  self.shape = args
 def forward(self, x):
  return x.view((x.size(0),)+self.shape)

以上这篇pytorch 在sequential中使用view来reshape的例子就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python获取Linux系统下的本机IP地址代码分享
Nov 07 Python
python监控网站运行异常并发送邮件的方法
Mar 13 Python
如何使用python爬取csdn博客访问量
Feb 14 Python
django数据库migrate失败的解决方法解析
Feb 08 Python
python实现朴素贝叶斯算法
Nov 19 Python
详解python算法之冒泡排序
Mar 05 Python
在Python中COM口的调用方法
Jul 03 Python
Python嵌套函数,作用域与偏函数用法实例分析
Dec 26 Python
keras获得model中某一层的某一个Tensor的输出维度教程
Jan 24 Python
Python3.7.0 Shell添加清屏快捷键的实现示例
Mar 23 Python
python中封包建立过程实例
Feb 18 Python
Pandas数据结构之Series的使用
Mar 31 Python
pytorch在fintune时将sequential中的层输出方法,以vgg为例
Aug 20 #Python
python实现证件照换底功能
Aug 20 #Python
pytorch多进程加速及代码优化方法
Aug 19 #Python
用Pytorch训练CNN(数据集MNIST,使用GPU的方法)
Aug 19 #Python
python opencv实现证件照换底功能
Aug 19 #Python
解决pytorch GPU 计算过程中出现内存耗尽的问题
Aug 19 #Python
将Pytorch模型从CPU转换成GPU的实现方法
Aug 19 #Python
You might like
PHP 全角转半角实现代码
2010/05/16 PHP
php excel类 phpExcel使用方法介绍
2010/08/21 PHP
3个PHP多维数组转为一维数组的方法实例
2014/03/13 PHP
采用thinkphp自带方法生成静态html文件详解
2014/06/13 PHP
PHP以mysqli方式连接类完整代码实例
2014/07/15 PHP
Eclipse的PHP插件PHPEclipse安装和使用
2014/07/20 PHP
Cygwin中安装PHP方法步骤
2015/07/04 PHP
javascript+dom树型菜单类,希望朋友们一起进步
2007/05/03 Javascript
JavaScript中判断函数是new还是()调用的区别说明
2011/04/07 Javascript
使用NodeJs 开发微信公众号(三)微信事件交互实例
2016/03/02 NodeJs
js实现多图左右切换功能
2016/08/04 Javascript
jQuery鼠标事件总结
2016/10/13 Javascript
Vue.js路由组件vue-router使用方法详解
2016/12/02 Javascript
轻松学习Javascript闭包
2017/03/01 Javascript
基于vuejs实现一个todolist项目
2017/04/11 Javascript
基于vue2.0实现的级联选择器
2017/06/09 Javascript
解决layer弹层遮罩挡住窗体的问题
2018/08/17 Javascript
详解ES6 Fetch API HTTP请求实用指南
2018/11/14 Javascript
微信小程序动态显示项目倒计时
2019/06/20 Javascript
Vue项目中使用flow做类型检测的方法
2020/03/18 Javascript
H5 js点击按钮复制文本到粘贴板
2020/11/19 Javascript
[00:06]Yes,it worked!小卡尔成功穿越时空加入战场!
2019/07/20 DOTA
Python数据拟合与广义线性回归算法学习
2017/12/22 Python
对Django中static(静态)文件详解以及{% static %}标签的使用方法
2019/07/28 Python
pygame库实现俄罗斯方块小游戏
2019/10/29 Python
实例教程 纯CSS3打造非常炫的加载动画效果
2014/11/05 HTML / CSS
怎样声明一个匿名的内部类
2016/06/01 面试题
中学老师的自我评价
2013/11/07 职场文书
《花的勇气》教后反思
2014/02/12 职场文书
班班通校本培训方案
2014/03/12 职场文书
机械电子工程专业求职信
2014/06/22 职场文书
商业用房租赁协议书
2014/10/13 职场文书
《一面五星红旗》教学反思
2016/02/23 职场文书
高中16字霸气押韵班级口号集锦!
2019/06/27 职场文书
创业项目大全(适合在家创业的项目)
2019/08/15 职场文书
Python超详细分步解析随机漫步
2022/03/17 Python