本文共 1343 字，大约阅读时间需要 4 分钟。

一、class torch.nn.ModuleList(modules=None)[source]

• 将submodules保存在一个 list 中。
• ModuleList可以像一般的Python list一样被索引。而且ModuleList中包含的modules已经被正确的注册，对所有的module method可见。
• 可以向python的列表一样使用append函数

二、参数说明:

• modules (list, optional) —— 将要被添加到MuduleList中的 modules 列表
• 类型：列表list

三、例子

import torch.nn as nnclass MyModule(nn.Module):    def __init__(self):        super(MyModule, self).__init__()        self.linears = nn.ModuleList([nn.Linear(10, 10) for i in range(10)])    def forward(self, x):        # ModuleList can act as an iterable, or be indexed         using ints        for i, l in enumerate(self.linears):            x = self.linears[i // 2](x) + l(x)        return x

• 输出

MyModule(  (linears): ModuleList(    (0): Linear(in_features=10, out_features=10, bias=True)    (1): Linear(in_features=10, out_features=10, bias=True)    (2): Linear(in_features=10, out_features=10, bias=True)    (3): Linear(in_features=10, out_features=10, bias=True)    (4): Linear(in_features=10, out_features=10, bias=True)    (5): Linear(in_features=10, out_features=10, bias=True)    (6): Linear(in_features=10, out_features=10, bias=True)    (7): Linear(in_features=10, out_features=10, bias=True)    (8): Linear(in_features=10, out_features=10, bias=True)    (9): Linear(in_features=10, out_features=10, bias=True)  ))

四、动态修改网络

• 中间（涉及列表的访问方式）、末尾都行。
• 还可以插入nn.Sequential对象。

转载地址：http://fbugf.baihongyu.com/