当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch ParameterList用法及代码示例


本文简要介绍python语言中 torch.nn.ParameterList 的用法。

用法:

class torch.nn.ParameterList(parameters=None)

参数

parameters(可迭代的,可选的) -要添加的 Parameter 的迭代

将参数保存在列表中。

ParameterList 可以像常规 Python 列表一样进行索引,但它包含的参数已正确注册,并且所有 Module 方法都可见。

例子:

class MyModule(nn.Module):
    def __init__(self):
        super(MyModule, self).__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

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.ParameterList。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。