快捷方式

参数列表

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
append(value)[source]

在列表的末尾追加给定的值。

参数

value (Any) – 要追加的值

返回值类型

参数列表

extend(values)[source]

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

参数

values (iterable) – 要追加的值的可迭代对象

返回值类型

自身

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源