PyTorchで独自のニューラルネットワーク層を作成する際、継承すべき基底クラスはどれか。

すべてのニューラルネットワークモジュールはnn.Moduleを継承する必要がある。