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

What does self.model(batch) do in pytorch?

我目前正在将 PyTorch 代码重写为 tensorflow。在这期间我发现了一行我不明白,所以我无法将它翻译成tensorflow。

谁能解释一下这是什么意思?

self.model(batch)

似乎self.model是一个layer/layers的神经网络,源自nn.Moduleclass。
调用 self.model(batch) 使用参数 batch.
调用 self.model__call__ 方法 如果您仔细检查,nn.Module.__call__ 会做一些“簿记”,但本质上,它会调用 self.modelforward 函数。