参数列表¶
- class torch.nn.ParameterList(values=None)[源代码]¶
以列表形式保存参数。
ParameterList可以像普通的 Python 列表一样使用，但作为Parameter的张量会被正确注册，并且所有Module方法都能看到它们。请注意，构造函数、分配列表中的元素、
append()方法和extend()方法会将任何Tensor转换为Parameter。- Parameters
参数 (可迭代对象, 可选) – 要添加到列表中的元素的可迭代对象。
示例：
class MyModule(nn.Module): def __init__(self): super().__init__() self.params = nn.ParameterList([nn.Parameter(torch.randn(10, 10)) for i in range(10)]) def forward(self, x): # ParameterList 可以作为可迭代对象，或者使用整数进行索引 for i, p in enumerate(self.params): x = self.params[i // 2].mm(x) + p.mm(x) return x
