PyTorch學習筆記之神經網絡包 nn 和優化器 optim

torch.nn 是專門爲神經網絡設計的模塊化接口。構建於 Autograd 之上，可用來定義和運行神經網絡。下面介紹幾個常用的類： 注： torch.nn 爲了方便使用，將它設置成 nn 的別名。 除了 nn 別名以外，我們還引用了 nn.functional，這個包中包含了神經網絡中使用的一些常用的函數，這些函數的特點是，不具有可學習的參數（ReLu, pool, DropOut 等），這些函

>>阅读原文<<