为何PyTorch中model(X_train)会自动调用forward函数？

问题 #

执行y_pred = model(X_train)时，未明确指定调用函数，为何会自动触发forward函数？是否因为模型类仅包含forward这一个函数？

解答 #

这和模型类中函数的数量无关，核心原因是PyTorch的nn.Module基类重载了__call__魔术方法：

• 当你以model(X_train)的方式调用模型实例时，实际调用的是实例的__call__方法，而非直接调用forward
• nn.Module的__call__方法内部会自动执行你定义的forward函数，同时还会处理额外的框架逻辑：
  • 触发预注册的钩子（hook），支持在模型执行流程中插入自定义操作
  • 维护计算图的构建，为后续反向传播（loss.backward()）做准备

哪怕你的模型类中定义了其他自定义方法，只要继承自nn.Module，调用实例时依然会通过__call__自动触发forward方法。

内容的提问来源于stack exchange，提问作者Limmi