模块列表¶
- class torch.nn.ModuleList(modules=None)[source]¶
在列表中保存子模块。
ModuleList
可以像普通的 Python 列表一样被索引,但它包含的模块被正确注册,并将对所有Module
方法可见。- 参数
modules (可迭代对象, 可选) – 要添加的模块的可迭代对象
示例
class MyModule(nn.Module): def __init__(self) -> None: super().__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