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将MongoDB里的ObjectId转换为时间戳的方法
Mar 13 Python
Python使用lxml模块和Requests模块抓取HTML页面的教程
May 16 Python
Python使用defaultdict读取文件各列的方法
May 11 Python
Python爬虫之xlml解析库(全面了解)
Aug 08 Python
Python实现获取nginx服务器ip及流量统计信息功能示例
May 18 Python
Django实战之用户认证(初始配置)
Jul 16 Python
pygame游戏之旅 添加游戏暂停功能
Nov 21 Python
Pycharm+Python+PyQt5使用详解
Sep 25 Python
python 单线程和异步协程工作方式解析
Sep 28 Python
python 如何实现遗传算法
Sep 22 Python
用python批量移动文件
Jan 14 Python
python中mongodb包操作数据库
Apr 19 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
解决PHP4.0 和 PHP5.0类构造函数的兼容问题
2013/08/01 PHP
PHP上传文件时自动分配路径的方法
2015/01/09 PHP
PHP正则判断一个变量是否为正整数的方法
2019/02/27 PHP
php和vue配合使用技巧和方法
2019/05/09 PHP
javascript实现仿银行密码输入框效果的代码
2007/12/13 Javascript
PNG背景在不同浏览器下的应用
2009/06/22 Javascript
关于IE、Firefox、Opera页面呈现异同 写脚本很痛苦
2009/08/28 Javascript
JQuery对表格进行操作的常用技巧总结
2014/04/23 Javascript
javascript编写实用的省市选择器
2015/02/12 Javascript
vue2.0项目集成Cesium的实现方法
2019/07/30 Javascript
java实现单链表增删改查的实例代码详解
2019/08/30 Javascript
react MPA 多页配置详解
2019/10/18 Javascript
解决Angularjs异步操作后台请求用$q.all排列先后顺序问题
2019/11/29 Javascript
JavaScript实现PC端横向轮播图
2020/02/07 Javascript
手把手教您实现react异步加载高阶组件
2020/04/07 Javascript
[52:10]LGD vs Optic Supermajor小组赛D组胜者组决赛 BO3 第二场 6.3
2018/06/04 DOTA
使用Python的Twisted框架编写非阻塞程序的代码示例
2016/05/25 Python
Python中pandas dataframe删除一行或一列:drop函数详解
2018/07/03 Python
selenium+python实现1688网站验证码图片的截取功能
2018/08/14 Python
Python 访问限制 private public的详细介绍
2018/10/16 Python
python 遍历列表提取下标和值的实例
2018/12/25 Python
Python3.4学习笔记之常用操作符,条件分支和循环用法示例
2019/03/01 Python
python使用matplotlib画柱状图、散点图
2019/03/18 Python
浅谈python 中类属性共享的问题
2019/07/02 Python
用python3 返回鼠标位置的实现方法(带界面)
2019/07/05 Python
详解用python计算阶乘的几种方法
2019/08/14 Python
python是否适合网页编程详解
2019/10/04 Python
Python大数据之从网页上爬取数据的方法详解
2019/11/16 Python
美国电力供应商店/电气批发商:USESI
2018/10/12 全球购物
CheapTickets泰国:廉价航班,查看促销价格并预订机票
2019/12/28 全球购物
自考生毕业自我鉴定
2013/10/10 职场文书
2014年民政局关于保密工作整改措施
2014/09/19 职场文书
科技活动周标语
2014/10/08 职场文书
病人慰问信范文
2015/02/15 职场文书
Redis调用Lua脚本及使用场景快速掌握
2022/03/16 Redis
微前端qiankun改造日渐庞大的项目教程
2022/06/21 Javascript