torch.nn.ModuleList

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

在列表中保持子模块。ModelList可以像正常Python列表一样检索，但是它所包含的模块已正确注册，并且对所有模块方法都可见。

参数：

• modules (iterable, optional) – an iterable of modules to add

例：

代码语言：javascript复制

class 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

append(module)[source]

Appends a given module to the end of the list.

参数：

• module (nn.Module) – 要附加的模块

extend(modules)[source]

将Python可迭代器中的模块追加到列表的末尾。

参数：

• modules (iterable) – 要追加的模块的可迭代

insert(index, module)[source]

在列表中给定索引之前插入给定模块。

参数：

• index (int) – 索引插入。
• module (nn.Module) – 模块插入

编程算法 python jquery

0 人点赞