? ? ? 在PyTorch源碼的torch/nn/modules/module.py文件中，有一條__call__語句和一條forward語句，如下：

__call__ : Callable[…, Any] = _call_impl
forward: Callable[…, Any] = _forward_unimplemented

? ? ? 在PyTorch中nn.Module類是所有神經(jīng)網(wǎng)絡(luò)模塊的基類，你的網(wǎng)絡(luò)也應(yīng)該繼承這個(gè)類，需要重載__init__和forward函數(shù)。以下是仿照PyTorch中Module和AlexNet類實(shí)現(xiàn)寫的假的實(shí)現(xiàn)的測(cè)試代碼：

from typing import Callable, Any, Listdef _forward_unimplemented(self, *input: Any) -> None:"Should be overridden by all subclasses"print("_forward_unimplemented")raise NotImplementedErrorclass Module:def __init__(self):print("Module.__init__")forward: Callable[..., Any] = _forward_unimplementeddef _call_impl(self, *input, **kwargs):print("Module._call_impl")result = self.forward(*input, **kwargs)return result__call__: Callable[..., Any] = _call_impldef cpu(self):print("Module.cpu")class AlexNet(Module):def __init__(self):print("AlexNet.__init__")super(AlexNet, self).__init__()def forward(self, x):print("AlexNet.forward")return xmodel = AlexNet()
x: List[int] = [1, 2, 3, 4]
print("result:", model(x))model.cpu()print("test finish")

? ? ? 執(zhí)行model(x)語句時(shí)，會(huì)調(diào)用AlexNet的forward函數(shù)，是因?yàn)锳lexNet的父類Module中的__call__函數(shù)：首先Module中有__call__方法，因此model(x)這條語句可以正常執(zhí)行。Module中并沒有直接給出__call__的實(shí)現(xiàn)體，而是__call__后緊跟冒號(hào)，此冒號(hào)表示類型注解；后面的Callable和Any是typing模塊中的，Callable表示可調(diào)用類型，即等號(hào)右邊應(yīng)該是一個(gè)可調(diào)用類型，此處指的是_call_impl；Any是一種特殊的類型，它與所有類型兼容；Callable[…, Any]表示_call_impl可接受任意數(shù)量的參數(shù)并返回Any。這里__call__實(shí)際指向了_call_impl函數(shù)，因此調(diào)用__call__實(shí)際是調(diào)用_call_impl。

? ? ? typing模塊的介紹參考：https://blog.csdn.net/fengbingchun/article/details/122288737

? ? ? _call_impl函數(shù)體內(nèi)會(huì)調(diào)用forward，Module中的forward的實(shí)現(xiàn)方式與__call__相同，但是_forward_unimplemented函數(shù)并沒有實(shí)現(xiàn)體，調(diào)用它會(huì)觸發(fā)Error即NotImplementedError。因此在子類AlexNet中一定要給出forward的具體實(shí)現(xiàn)，否則調(diào)用的將是_forward_unimplemented。

? ? ? 測(cè)試代碼執(zhí)行結(jié)果如下：

? ? ? 如果注釋掉AlexNet中的forward，則執(zhí)行結(jié)果如下：

? ? ? GitHub：?https://github.com/fengbingchun/PyTorch_Test

總結(jié)

以上是生活随笔為你收集整理的PyTorch中nn.Module类中__call__方法介绍的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。