torch.nn.Module.named_parameters ()

参考 torch.nn.Module.named_parameters () - 云+社区 - 腾讯云

named_parameters(prefix='', recurse=True)[source]

返回模块参数上的迭代器，生成参数的名称和参数本身。

参数：

• prefix (str) – 在所有参数名称前加上前缀。

• recurse (bool) – 如果为真，则生成此模块和所有子模块的参数。否则，只生成此模块的直接成员的参数。

生产量：

(string, Parameter) – 包含名称和参数的元组

例：

>>> for name, param in self.named_parameters():
>>>    if name in ['bias']:
>>>        print(param.size())