1 兩種等價(jià)的構(gòu)造方式

1.1 構(gòu)建Net類：

1.2 使用torch.nn.Sequential

nn.Sequential里面的模塊按照順序進(jìn)行排列的，所以必須確保前一個(gè)模塊的輸出大小和下一個(gè)模塊的輸入大小是一致的。

這兩種表達(dá)方式唯一的不同就是，在1.2中的方法中，激活函數(shù)也獨(dú)立算作一層。

2?OrderedDict

nn.Sequential中可以使用OrderedDict來指定每個(gè)module的名字，而不是采用默認(rèn)的命名方式(按序號(hào) 0,1,2,3...)。

總結(jié)

以上是生活随笔為你收集整理的pytorch 学习笔记：nn.Sequential构造神经网络的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。