pytorch 中forward 的用法與解釋說明
更新時間：2021年02月26日 11:32:03 作者：JY丫丫
這篇文章主要介紹了pytorch 中forward 的用法與解釋說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
前言
最近在使用pytorch的時候，模型訓練時，不需要使用forward，只要在實例化一個對象中傳入對應的參數(shù)就可以自動調用 forward 函數(shù)
即：
forward 的使用
class Module(nn.Module): def __init__(self): super(Module, self).__init__() # ...... def forward(self, x): # ...... return x data = ..... #輸入數(shù)據(jù) # 實例化一個對象 module = Module() # 前向傳播 module(data) # 而不是使用下面的 # module.forward(data)
實際上
module(data)
是等價于
module.forward(data)
forward 使用的解釋
等價的原因是因為 python calss 中的__call__和__init__方法.
class A():
def __call__(self):
print('i can be called like a function')
a = A()
a()
out：
i can be called like a function
__call__里調用其他的函數(shù)
class A():
def __call__(self, param):
print('i can called like a function')
print('傳入參數(shù)的類型是：{} 值為： {}'.format(type(param), param))
res = self.forward(param)
return res
def forward(self, input_):
print('forward 函數(shù)被調用了')
print('in forward, 傳入參數(shù)類型是：{} 值為: {}'.format( type(input_), input_))
return input_
a = A()
input_param = a('i')
print("對象a傳入的參數(shù)是：", input_param)
out:
i can called like a function
傳入參數(shù)的類型是：<class ‘str'> 值為： i
forward 函數(shù)被調用了
in forward, 傳入參數(shù)類型是：<class ‘str'> 值為: i
對象a傳入的參數(shù)是： i
補充：Pytorch 模型中nn.Model 中的forward() 前向傳播不調用 解釋
在pytorch 中沒有調用模型的forward（）前向傳播，只實列化后把參數(shù)傳入。
定義模型
class Module(nn.Module): def __init__(self): super(Module, self).__init__() # ...... def forward(self, x): # ...... return x data = ..... #輸入數(shù)據(jù) # 實例化一個對象 module = Module() # 前向傳播 直接把輸入傳入實列化 module(data) #沒有使用module.forward(data)
實際上module(data) 等價于module.forward(data)
等價的原因是因為 python calss 中的__call__ 可以讓類像函數(shù)一樣調用
當執(zhí)行model(x)的時候，底層自動調用forward方法計算結果
class A():
def __call__(self):
print('i can be called like a function')
a = A()
a()
>>>i can be called like a function
在__call__ 里可調用其它的函數(shù)
class A():
def __call__(self, param):
print('我在__call__中，傳入參數(shù)',param)
res = self.forward(param)
return res
def forward(self, x):
print('我在forward函數(shù)中，傳入參數(shù)類型是值為: ',x)
return x
a = A()
y = a('i')
>>> 我在__call__中，傳入參數(shù) i
>>>我在forward函數(shù)中，傳入參數(shù)類型是值為: i
print("傳入的參數(shù)是：", y)
>>>傳入的參數(shù)是： i
以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關文章
現(xiàn)成的YOLOv5代碼真的很香,不管口碑怎么樣,我用著反正是挺爽的,下面這篇文章主要給大家介紹了關于yolov5中head修改為decouple?head的相關資料,需要的朋友可以參考下2022-06-06
今天小編就為大家分享一篇Python之——生成動態(tài)路由軌跡圖的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11
今天小編就為大家分享一篇pytorch中的embedding詞向量的使用方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
在 Python 中,邊遍歷邊刪除列表元素通常是一個不推薦的操作,因為它會改變列表的大小,可能會導致一些意料之外的行為,例如,元素被刪除后,列表的索引會發(fā)生變化,可能導致漏掉某些元素或者遍歷到錯誤的位置,所以本文介紹了Python邊遍歷邊刪除列表元素的幾種方法2024-12-12
這篇文章主要介紹了Python pytest.main()運行測試用例，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習吧2022-12-12
大家都知道視頻就是一幀一幀的圖片構成的。那么想要實現(xiàn)將視頻轉換為字符畫，就要將一部視頻全部逐幀拆解成圖片，然后采取和以前相同的將圖片轉換為字符畫的算法即可。本文將詳細為大家介紹如何實現(xiàn)，需要的可以參考一下2021-12-12
這篇文章主要介紹了Python黑魔法遠程控制開機的實例,文中有非常詳細的代碼示例,對正在學習python的小伙伴們有很大的幫助,需要的朋友可以參考下2021-04-04
Python自動修改電腦靜態(tài)IP地址的實現(xiàn)示例
通過Python自動修改電腦的靜態(tài)IP地址可以極大地提高我們的工作效率,減少手動修改IP地址帶來的錯誤,本文就來介紹一下Python自動修改電腦靜態(tài)IP地址的實現(xiàn)示例,感興趣的可以了解一下2023-11-11
這篇文章主要介紹了Python Matplotlib庫入門指南,本文講解了Matplotlib是什么,然后給出了Matplotlib基礎繪圖實例如繪制折線圖、繪制多線圖,并給出了圖例功能使用實例,需要的朋友可以參考下2015-05-05
在本篇文章里小編給大家整理了是關于學python安裝什么軟件的相關知識點內容，有需要的朋友們學習下。2019-10-10