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編程開發(fā)時(shí)間序列calendar模塊示例詳解
這篇文章主要為大家介紹了python編程開發(fā)時(shí)間序列calendar模塊示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助祝大家多多進(jìn)步早日升職加薪2021-11-11
安裝Keras,tensorflow,并實(shí)現(xiàn)將虛擬環(huán)境添加到j(luò)upyter?notebook
這篇文章主要介紹了安裝Keras,tensorflow,并實(shí)現(xiàn)將虛擬環(huán)境添加到j(luò)upyter?notebook,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-03-03
文章介紹了Python3.x中的關(guān)鍵字參數(shù)特性,說(shuō)明了如何在函數(shù)定義中使用*符號(hào)來(lái)限制參數(shù)只能通過(guò)關(guān)鍵字傳遞,而不是位置,這種特性在處理參數(shù)較多或順序可能導(dǎo)致混淆的函數(shù)時(shí)特別有用,提高了代碼的可讀性和健壯性2024-11-11
完美解決keras保存好的model不能成功加載問(wèn)題
這篇文章主要介紹了完美解決keras保存好的model不能成功加載問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-06-06
python循環(huán)接收http請(qǐng)求數(shù)據(jù)方式
這篇文章主要介紹了python循環(huán)接收http請(qǐng)求數(shù)據(jù)方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-06-06
Django中ORM表的創(chuàng)建和增刪改查方法示例
這篇文章主要給大家介紹了關(guān)于Django中ORM表的創(chuàng)建和增刪改查等基本操作的方法，還給大家分享了django orm常用查詢篩選的相關(guān)內(nèi)容，分享出來(lái)供大家參考學(xué)習(xí)，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧。2017-11-11
巧用Python實(shí)現(xiàn)自動(dòng)化翻譯任務(wù)分配詳解
這篇文章主要為大家詳細(xì)介紹了如何使用Python實(shí)現(xiàn)自動(dòng)化翻譯任務(wù)分配,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2025-11-11
Python設(shè)計(jì)模式編程中解釋器模式的簡(jiǎn)單程序示例分享
這篇文章主要介紹了Python設(shè)計(jì)模式編程中解釋器模式的簡(jiǎn)單程序示例分享,解釋器模式強(qiáng)調(diào)用抽象類來(lái)表達(dá)程序中將要實(shí)現(xiàn)的功能,需要的朋友可以參考下2016-03-03
Python實(shí)現(xiàn)簡(jiǎn)易凱撒密碼的示例代碼
密碼的使用最早可以追溯到古羅馬時(shí)期，《高盧戰(zhàn)記》有描述愷撒曾經(jīng)使用密碼來(lái)傳遞信息，即所謂的“愷撒密碼”。本文將利用Python實(shí)現(xiàn)簡(jiǎn)易的凱撒密碼，感興趣的可以了解一下2022-09-09
這篇文章主要介紹了python tkinter事件綁定，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03