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里面的模型为子模块。这里只是创建了一层网络。

conv2d官方文档

 

posted @ 2022-09-11 16:26  海_纳百川  阅读(220)  评论(0编辑  收藏  举报
本站总访问量