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

更新时间：2019年08月20日 08:54:52 作者：青盏

今天小编就为大家分享一篇pytorch 在sequential中使用view来reshape的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

解决matplotlib库show()方法不显示图片的问题
今天小编就为大家分享一篇解决matplotlib库show()方法不显示图片的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-05-05
对Python3 序列解包详解
今天小编就为大家分享一篇对Python3 序列解包详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-02-02
python的图形用户界面介绍
大家好，本篇文章主要讲的是python的图形用户界面介绍，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览
2022-01-01
Python可跨平台实现获取按键的方法
这篇文章主要介绍了Python可跨平台实现获取按键的方法,分别针对windows及Unix等不同平台获取按键的技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-03-03
python3+django2开发一个简单的人员管理系统过程详解
这篇文章主要介绍了python3+django2开发一个简单的人员管理系统过程详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-07-07
python批量爬取图片的方法详解
这篇文章给大家介绍了如何使用python批量爬取图片,文中通过代码示例给大家介绍的非常详细,对大家的学习或工作有一定的帮助,需要的朋友可以参考下
2023-12-12
python使用Paramiko模块实现远程文件拷贝
这篇文章主要为大家详细介绍了python使用Paramiko模块实现远程文件拷贝，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-04-04
mAP计算目标检测精确度实现源码
这篇文章主要为大家介绍了mAP计算目标检测精确度的实现源码，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-05-05
Python+Dlib+Opencv实现人脸采集并表情判别功能的代码
这篇文章主要介绍了Python+Dlib+Opencv实现人脸采集并表情判别,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-07-07
Spring http服务远程调用实现过程解析
这篇文章主要介绍了Spring http服务远程调用实现过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-06-06