Pytorch中容器nn.Sequential(*layers)
网上关于pytorch中容器的博客和专栏有很多,推荐一下两个网址:
1.https://blog.csdn.net/u013548568/article/details/80294708
2.https://zhuanlan.zhihu.com/p/183093644
之前见过一段少见的代码:
layers = []
layers += [nn.Linear(118,60)]
layers += [nn.Tanh()]
layers += [nn.Linear(60,30)]
layers += [nn.Tanh()]
layers += [nn.Linear(30,10)]
layers += [nn.Tanh()]
layers += [nn.Linear(10,1)]
self.encoder = nn.Sequential(*layers)
其实它的同义代码如下:
self.encoder = nn.Sequential(
nn.Linear(118, 60),
nn.Tanh(),
nn.Linear(60, 30),
nn.Tanh(),
nn.Linear(30, 10),
nn.Tanh(),
nn.Linear(10, 1),)