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]))