参数列表¶
- class torch.nn.ParameterList(values=None)[source]¶
在列表中保存参数。
ParameterList
可以像常规 Python 列表一样使用,但作为Parameter
的张量将被正确注册,并且所有Module
方法都可见。请注意,构造函数、赋值列表元素、
append()
方法和extend()
方法将任何Tensor
转换为Parameter
。- 参数
parameters (iterable, optional) – 要添加到列表中的元素的可迭代对象。
示例
class MyModule(nn.Module): def __init__(self) -> None: super().__init__() self.params = nn.ParameterList([nn.Parameter(torch.randn(10, 10)) for i in range(10)]) def forward(self, x): # ParameterList can act as an iterable, or be indexed using ints for i, p in enumerate(self.params): x = self.params[i // 2].mm(x) + p.mm(x) return x