欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 财经 > 金融 > Pytorch笔记(1-5)

Pytorch笔记(1-5)

2024/10/24 9:18:11 来源:https://blog.csdn.net/2301_81989837/article/details/140563473  浏览:    关键词:Pytorch笔记(1-5)

张量

张量可以理解为numpy里面的矩阵,可以进行矩阵的操作

张量初始化

x_data = torch.tensor([[1,2],[2,3]])

模型的建立

首先建立一个单元,在pytorch里面,一个单元就是神经网络的一个节点,前向传播。

一般以一个类定义

class LearnModel(torch.nn.Module):def __init__(self):super(LinearModel,self).__init__()# 必须要有self.linear = torch.nn.Linear(1,1)# 接受两个参数一个是输入数据的维度,一个是输出数据的维度,这个函数就是求出参数w和b再顺便做好相乘加bdef forward(self,x):y_pred = self.linear(x)# 前向传播return y_pred

损失函数和参数更新


criterion = torch.nn.MSELoss(size_average = True)
# 损失函数,均方根损失函数,接受两个参数,上面这个为True时求均值
optimizer = torch.optim.SGD(model.parameters(), lr = 0.01)
# 这是一个优化器,就是更新参数的,第一个参数就是你要更新哪些参数,第一个函数可以选择所有的参数,lr就是learn_rate学习率

线性回归


model = LinearModel()
for epoch in range(100):y_pred = model(x_data) # 前向传播获得y的预测值loss = criterion(y_pred, y_data) # 求出代价optimizer.zero_grad() # 清零梯度loss.backward() # 自动计算梯度optimizer.step() # 更新w和b的值

版权声明:

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

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