PyTorch Lightning中调用self(x)的含义及可调用性解析

先看你给出的这段PyTorch Lightning模型的训练步骤代码：

def training_step(self, batch, batch_idx):
    x, y = batch
    y_hat = self(x)
    loss = F.cross_entropy(y_hat, y)
    return loss

1. 调用self(x)的含义 #

这里的self就是当前的模型实例本身，self(x)等价于触发模型的前向传播流程——把输入数据x喂进模型，经过你定义的网络层计算后，得到预测结果y_hat。

2. 为什么self可以作为可调用对象 #

Python里的类实例要能像函数一样被调用，只需要在类中实现__call__魔法方法。你的模型类肯定继承了PyTorch的nn.Module（PyTorch Lightning的基类也基于它），而nn.Module已经内置实现了__call__方法。当你调用self(x)时，会触发这个__call__方法，它不仅会执行你在模型里定义的forward方法（真正的前向计算逻辑），还会自动处理梯度跟踪、钩子函数执行等框架层面的额外操作。

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