pytorch中的nn.ModuleList(list)
self.convs = nn.ModuleList([nn.Conv2d(1, config.num_filters, (k, config.embed)) for k in (2,3,4)])
在这里list是多个model,可以理解成子模块。
ModuleList的作用:不是创建三层前后连接的网络,而是创建三个上下并列的网络,所以称model里面的模型为子模块。这里只是创建了一层网络。
本文来自博客园,作者:海_纳百川,转载请注明原文链接:https://www.cnblogs.com/chentiao/p/16684291.html,如有侵权联系删除