pytorch中forward的理解

左耳听风發表於2024-05-21

使用pytorch的時候,模型訓練時,不需要使用forward,只要在例項化一個物件中傳入對應的引數就可以自動呼叫 forward 函式

pytorch中forward的理解
 1 class Module(nn.Module):
 2     def __init__(self):
 3         super(Module, self).__init__()
 4         # ......
 5        
 6     def forward(self, x):
 7         # ......
 8         return x
 9 
10 data = .....  #輸入資料
11 # 例項化一個物件
12 module = Module()
13 # 前向傳播
14 module(data)  
15 # 而不是使用下面的
16 # module.forward(data)   
View Code

在宣告網路架構是,常常使用class RegionProposalNetwork(nn.Module),其中nn.Module中包含了__call__函式,在函式中呼叫了forward,由於繼承關係,對於RegionProposalNetwork同樣具備__call__函式的功能。

相關文章