快捷導(dǎo)航

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):
  # 如果數(shù)據(jù)集最后一個batch樣本數(shù)量小于定義的batch_batch大小，會出現(xiàn)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的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python實現(xiàn)代碼審查自動回復(fù)消息
這篇文章主要介紹了python實現(xiàn)代碼審查回復(fù)消息生成的示例，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下
2021-02-02
Python實現(xiàn)批量讀取HDF多波段柵格數(shù)據(jù)并繪制像元直方圖
這篇文章主要為大家詳細介紹了如何基于Python語言gdal模塊，實現(xiàn)多波段HDF柵格圖像文件的讀取、處理與像元值可視化（直方圖繪制）等操作，需要的可以參考一下
2023-03-03
python逐行讀取文件內(nèi)容的三種方法
這篇文章主要介紹了python逐行讀取文件內(nèi)容的三種方法，非常的簡單，下面直接看代碼吧
2014-01-01
python ffmpeg任意提取視頻幀的方法
這篇文章主要介紹了python ffmpeg任意提取視頻幀的方法，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2020-02-02
通過python實現(xiàn)彈窗廣告攔截過程詳解
這篇文章主要介紹了通過python實現(xiàn)彈窗廣告攔截過程詳解,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2019-07-07
python可擴展的Blender 3D插件開發(fā)匯總
這篇文章主要為大家介紹了python可擴展的Blender 3D插件開發(fā)匯總,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
2023-09-09
python collections模塊示例詳解
Python的collections模塊是其標準庫中的一部分,包含了一些擴展內(nèi)置數(shù)據(jù)類型的有用數(shù)據(jù)結(jié)構(gòu),如OrderedDict、defaultdict、Counter、deque和namedtuple等,這些數(shù)據(jù)結(jié)構(gòu)在不同場景下都非常有用,能提供更高效的數(shù)據(jù)操作和方便的數(shù)據(jù)訪問方式,有助于提升代碼的性能和可讀性
2024-10-10
新手該如何學(xué)python怎么學(xué)好python？
怎么學(xué)好python？怎么靈活應(yīng)用python？
2008-10-10
Python Flask框架擴展操作示例
這篇文章主要介紹了Python Flask框架擴展操作,結(jié)合實例形式分析了Python Flask-Script擴展命令行操作實現(xiàn)方法,需要的朋友可以參考下
2019-05-05
Python使用pymssql連接SQL?SEVER數(shù)據(jù)庫全流程
SQL Server是微軟推出的重量級的數(shù)據(jù)庫,目前有多個版本,如2000、2008、2012等,下面這篇文章主要給大家介紹了關(guān)于Python使用pymssql連接SQL?SEVER數(shù)據(jù)庫的相關(guān)資料,需要的朋友可以參考下
2023-12-12