使用pytorch的時候,模型訓練時,不需要使用forward,只要在例項化一個物件中傳入對應的引數就可以自動呼叫 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)
在宣告網路架構是,常常使用class RegionProposalNetwork(nn.Module)
,其中nn.Module
中包含了__call__
函式,在函式中呼叫了forward
,由於繼承關係,對於RegionProposalNetwork同樣具備__call__
函式的功能。