self.model(batch) 在 pytorch 中做什么？

Question

我目前正在将 PyTorch 代码重写为张量流。在此过程中，我发现了一行我不明白的内容，因此我无法将其翻译为张量流。

有人可以解释一下这是什么意思吗？

self.model(batch)

Answer 1

它看起来像是self.model从类派生的神经网络的一个或多个层nn.Module。
该调用使用参数 来self.model(batch)调用 的self.model方法。 如果您仔细检查，会进行一些“簿记”，但本质上，它会调用's函数。__call__batch
nn.Module.__call__self.modelforward