pytorch - 在pytorch中，损失函数需要继承nn.module的情况有哪些？

Question

我对 PyTorch 中的损失函数感到困惑。有些人将损失函数定义为普通的python函数，而另一些人通过定义一个继承nn.Module的类来定义损失函数。所以想知道什么情况下我们需要通过继承nn.Module来定义损失函数？非常感谢。

Answer 1

一般来说，nn.Module只有当你想在这个模块中有可训练的变量时，才需要继承，否则继承它是可选的。

所以同样适用于损失函数，如果它不包含这样的变量（我认为这是主要情况），则不需要继承。