参数列表

类torch.nn.ParameterList(values=None)[源代码]

将参数保存在列表中。

ParameterList 可以像普通的 Python 列表一样使用，但被标记为 Parameter 的张量会被正确注册，并且所有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)[源代码]

将一个给定的值添加到列表的末尾。

参数

value (Any) - 需要追加的值

返回类型

ParameterList

extend(values)[源代码]

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

参数

values (可迭代对象) – 需要追加的值的集合，该集合是可迭代的对象

返回类型