self.model(batch) 在 pytorch 中做什么?
What does self.model(batch) do in pytorch?
我目前正在将 PyTorch 代码重写为 tensorflow。在这期间我发现了一行我不明白,所以我无法将它翻译成tensorflow。
谁能解释一下这是什么意思?
self.model(batch)
似乎self.model
是一个layer/layers的神经网络,源自nn.Module
class。
调用 self.model(batch)
使用参数 batch
.
调用 self.model
的 __call__
方法
如果您仔细检查,nn.Module.__call__
会做一些“簿记”,但本质上,它会调用 self.model
的 forward
函数。
我目前正在将 PyTorch 代码重写为 tensorflow。在这期间我发现了一行我不明白,所以我无法将它翻译成tensorflow。
谁能解释一下这是什么意思?
self.model(batch)
似乎self.model
是一个layer/layers的神经网络,源自nn.Module
class。
调用 self.model(batch)
使用参数 batch
.
调用 self.model
的 __call__
方法
如果您仔细检查,nn.Module.__call__
会做一些“簿记”,但本质上,它会调用 self.model
的 forward
函数。