在Pytorch中使用 nn.Sequential()自動添加網絡層

 nn.Sequential()定義網絡簡單高效，可以寫死，可以自動添加add_module

參考鏈接：pytorch中的add_module函數 - 蒙面的普羅米修斯 - 博客園 (cnblogs.com)

                  pytorch nn.Sequential()動態添加方法 - 慢行厚積 - 博客園 (cnblogs.com)

 

self.encoder = nn.Sequential()
        for i in range(len(self.nH_enc)):
            if i == 0:
                self.encoder.add_module('layer_{}'.format(i), nn.Linear(self.input_dim, self.nH_enc[i]))
            else:
                self.encoder.add_module('layer_{}'.format(i), nn.Linear(self.nH_enc[i-1], self.nH_enc[i]))