pytorch中Module模块中named_parameters函数

最新推荐文章于 2025-03-12 17:49:39 发布

小白827

最新推荐文章于 2025-03-12 17:49:39 发布

阅读量6.7k

点赞数 2

分类专栏： pytorch

本文链接：https://blog.youkuaiyun.com/qq_24503095/article/details/103630382

版权

本文介绍PyTorch中Module模块的named_parameters()函数，该函数用于获取模型中所有参数的名称及其对应的数值，对于理解和调试深度学习模型的参数非常有用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

函数named_parameters()，返回各层中参数名称和数据

class MLP(nn.Module):
    def __init__(self):
        super(MLP, self).__init__()
        self.hidden = nn.Sequential(
            nn.Linear(256,64),
            nn.ReLU(inplace=True),
            nn.Linear(64,10)
                )
    def forward(self, x):
        return self.hidden(x)

    
#x = torch.randperm(256*2).view(-1,256)
net = MLP()
net(x