参数列表 ¶

class torch.nn.ParameterList(values=None)[来源][来源] ¶

存储在列表中的参数。

ParameterList 可以像常规 Python 列表一样使用，但 Parameter 正确注册的 Tensors 将对所有 Module 方法可见。

注意，构造函数、分配列表元素的方法、 append() 方法以及 extend() 方法会将任何 Tensor 转换为 Parameter 。

参数:

参数 (iterable, 可选) – 要添加到列表中的元素的可迭代对象。

示例：

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

append(value)

在列表末尾追加给定值。

参数:

值（任何）- 要追加的值

返回类型:

参数列表

extend(values)

将 Python 可迭代对象的值追加到列表末尾。

参数:

values（可迭代）- 要追加的值的可迭代序列

返回类型:

自身