Pytorch笔记(1-5)

张量

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

张量初始化

python 复制代码
x_data = torch.tensor([[1,2],[2,3]])

模型的建立

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

一般以一个类定义

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

损失函数和参数更新

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

线性回归

python 复制代码
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的值
相关推荐
飞哥数智坊20 小时前
GPT-5-Codex 发布,Codex 正在取代 Claude
人工智能·ai编程
倔强青铜三20 小时前
苦练Python第46天:文件写入与上下文管理器
人工智能·python·面试
虫无涯21 小时前
Dify Agent + AntV 实战:从 0 到 1 打造数据可视化解决方案
人工智能
Dm_dotnet1 天前
公益站Agent Router注册送200刀额度竟然是真的
人工智能
算家计算1 天前
7B参数拿下30个世界第一!Hunyuan-MT-7B本地部署教程:腾讯混元开源业界首个翻译集成模型
人工智能·开源
机器之心1 天前
LLM开源2.0大洗牌:60个出局,39个上桌,AI Coding疯魔,TensorFlow已死
人工智能·openai
Juchecar1 天前
交叉熵:深度学习中最常用的损失函数
人工智能
林木森ai1 天前
爆款AI动物运动会视频,用Coze(扣子)一键搞定全流程(附保姆级拆解)
人工智能·aigc
聚客AI1 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
BeerBear1 天前
【保姆级教程-从0开始开发MCP服务器】一、MCP学习压根没有你想象得那么难!.md
人工智能·mcp