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的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python实现代码审查回复消息生成的示例，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下2021-02-02
Python实现批量读取HDF多波段栅格数据并绘制像元直方图
这篇文章主要为大家详细介绍了如何基于Python语言gdal模块，实现多波段HDF栅格图像文件的读取、处理与像元值可视化（直方图绘制）等操作，需要的可以参考一下2023-03-03
这篇文章主要介绍了python逐行读取文件内容的三种方法，非常的简单，下面直接看代码吧2014-01-01
这篇文章主要介绍了python ffmpeg任意提取视频帧的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-02-02
这篇文章主要介绍了通过python实现弹窗广告拦截过程详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
这篇文章主要为大家介绍了python可扩展的Blender 3D插件开发汇总,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-09-09
Python的collections模块是其标准库中的一部分,包含了一些扩展内置数据类型的有用数据结构,如OrderedDict、defaultdict、Counter、deque和namedtuple等,这些数据结构在不同场景下都非常有用,能提供更高效的数据操作和方便的数据访问方式,有助于提升代码的性能和可读性2024-10-10
怎么学好python？怎么灵活应用python？2008-10-10
这篇文章主要介绍了Python Flask框架扩展操作,结合实例形式分析了Python Flask-Script扩展命令行操作实现方法,需要的朋友可以参考下2019-05-05
Python使用pymssql连接SQL SEVER数据库全流程
SQL Server是微软推出的重量级的数据库,目前有多个版本,如2000、2008、2012等,下面这篇文章主要给大家介绍了关于Python使用pymssql连接SQL SEVER数据库的相关资料,需要的朋友可以参考下2023-12-12
最新评论