快捷導(dǎo)航

pytorch 在sequential中使用view來(lái)reshape的例子

更新時(shí)間：2019年08月20日 08:54:52 作者：青盞

今天小編就為大家分享一篇pytorch 在sequential中使用view來(lái)reshape的例子，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

pytorch中view是tensor方法，然而在sequential中包裝的是nn.module的子類，

因此需要自己定義一個(gè)方法：

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ù)集最后一個(gè)batch樣本數(shù)量小于定義的batch_batch大小，會(huì)出現(xiàn)mismatch問(wèn)題?？梢宰约盒薷南?，如只傳入后面的shape，然后通過(guò)x.szie(0)，來(lái)輸入。
  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來(lái)reshape的例子就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

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