快捷導(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)文章

PyTorch加載預(yù)訓(xùn)練模型實例(pretrained)
今天小編就為大家分享一篇PyTorch加載預(yù)訓(xùn)練模型實例(pretrained)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-01-01
Python中文編碼那些事
由于中文的特殊性，造成了在各大編程語言中都存在編碼轉(zhuǎn)換的問題，今天我們來探討下python中中文編碼如何處理呢
2014-06-06
使用python的pandas為你的股票繪制趨勢圖
這篇文章主要介紹了通過python為你的股票繪制趨勢圖,動手寫個小程序，把股票趨勢每天早上發(fā)到郵箱里，用 python 的 pandas, matplotlib 寫起來很容易，幾十行代碼搞定。,需要的朋友可以參考下
2019-06-06
Python中的tkinter庫簡單案例詳解
tkinter?是?Python?的標(biāo)準(zhǔn)?GUI?庫，Python?使用?tkinter?可以快速的創(chuàng)建?GUI?應(yīng)用程序，今天通過本文給大家分享Python中的tkinter庫簡單案例詳解,需要的朋友可以參考下
2022-01-01
Pycharm安裝PyQt5的詳細(xì)教程
PyCharm 是一款功能強大的 Python 編輯器，具有跨平臺性，鑒于目前最新版 PyCharm 使用教程較少，為了節(jié)約時間，來介紹一下Pycharm安裝PyQt5的詳細(xì)教程，感興趣的朋友跟隨小編一起看看吧
2021-12-12
OpenCV學(xué)習(xí)之圖像形態(tài)學(xué)處理詳解
這篇文章主要為大家詳細(xì)介紹了OpenCV中圖像形態(tài)學(xué)處理的相關(guān)知識，例如：腐蝕操作、膨脹操作、開閉運算、梯度運算、Top Hat Black Hat運算等操作，需要的可以參考一下
2023-02-02
Python對象中__del__方法起作用的條件詳解
今天小編就為大家分享一篇Python對象中__del__方法起作用的條件詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-11-11
django基于存儲在前端的token用戶認(rèn)證解析
這篇文章主要介紹了django基于存儲在前端的token用戶認(rèn)證解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2019-08-08
python lxml中etree的簡單應(yīng)用
這篇文章主要介紹了python lxml中etree的簡單應(yīng)用，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-05-05
利用pyinstaller打包exe文件的基本教程
這篇文章主要給大家介紹了關(guān)于利用pyinstaller打包exe文件的基本方法，文中通過示例代碼介紹的非常詳細(xì)，對大家學(xué)習(xí)或者使用pyinstaller具有一定的參考學(xué)習(xí)價值，需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧
2019-05-05