当前位置: 首页> 财经> 访谈 > 网站制作 网站建设怎么做的_秦皇岛开发区_关键词排名优化公司_百度关键词热度

网站制作 网站建设怎么做的_秦皇岛开发区_关键词排名优化公司_百度关键词热度

时间:2025/7/11 22:41:01来源:https://blog.csdn.net/m0_52062236/article/details/145882159 浏览次数:0次
网站制作 网站建设怎么做的_秦皇岛开发区_关键词排名优化公司_百度关键词热度

model.parameters()

import torch
import torch.nn as nnclass SimpleModel(nn.Module):def __init__(self):super(SimpleModel, self).__init__()self.fc1 = nn.Linear(10, 5)  # 输入维度为10,输出维度为5self.fc2 = nn.Linear(5, 2)   # 输入维度为5,输出维度为2def forward(self, x):x = torch.relu(self.fc1(x))x = self.fc2(x)return xmodel = SimpleModel()# 获取模型的参数
for param in model.parameters():print(param.shape, param.requires_grad)

假设打印结果:

torch.Size([5, 10]) True   # 第一层的权重
torch.Size([5]) True       # 第一层的偏置
torch.Size([2, 5]) True    # 第二层的权重
torch.Size([2]) True       # 第二层的偏置

线性层Linear WX+b

model.parameters() 返回一个生成器(generator),它会逐个生成模型中所有可训练的参数(Parameter 对象)。这些参数通常是张量(torch.Tensor),并且默认情况下它们的 requires_grad=True,这意味着它们会在反向传播中自动计算梯度

model.named_parameters()

for name, param in linearModel.named_parameters():print(name, param.grad)
# linear.weight tensor([[...]])
# linear.bias   tensor([[...]])
关键字:网站制作 网站建设怎么做的_秦皇岛开发区_关键词排名优化公司_百度关键词热度

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: