轉(zhuǎn)載：https://blog.csdn.net/xu380393916/article/details/97280035
前言
最近在使用pytorch的時候突诬，模型訓(xùn)練時掘猿，不需要調(diào)用forward，只要在實(shí)例化一個對象中傳入對應(yīng)的參數(shù)就可以自動調(diào)用forward函數(shù)商佑，即
forward的使用
class Module(nn.Module):
def __init__(self):
super(Module, self).__init__()
# ......
def forward(self, x):
# ......
return x
data = ..... #輸入數(shù)據(jù)
# 實(shí)例化一個對象
module = Module()
# 前向傳播
module(data)
# 而不是使用下面的
# module.forward(data)
實(shí)際上
module(data)
# 等價于
module.forward(data)
forward 使用的等價解釋
等價的原因是python class中的 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
實(shí)例二
class A():
def __call__(self, param):
print('i can called like a function')
print('傳入?yún)?shù)的類型是：{} 值為： {}'.format(type(param), param))
res = self.forward(param)
return res
def forward(self, input_):
print('forward 函數(shù)被調(diào)用了')
print('in forward, 傳入?yún)?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
傳入?yún)?shù)的類型是：<class ‘str’> 值為： i
forward 函數(shù)被調(diào)用了
in forward, 傳入?yún)?shù)類型是：<class ‘str’> 值為: i
對象a傳入的參數(shù)是： i
參考資料
(1條消息)pytorch 之 call, init,forward - Every moment of My life !!! - CSDN博客
https://blog.csdn.net/xxboy61/article/details/88101192
(1條消息)PyTorch之前向傳播函數(shù)forward - 鵲踏枝-碼農(nóng)的專欄 - CSDN博客
https://blog.csdn.net/u011501388/article/details/84062483
