Pytorch线性回归实现(原理)

设置梯度

直接在tensor中设置 requires_grad=True,每次操作这个数的时候,就会保存每一步的数据。也就是保存了梯度相关的数据。

python 复制代码
import torch
x = torch.ones(2, 2, requires_grad=True)  #初始化参数x并设置requires_grad=True用来追踪其计算历史
print(x)
#tensor([[1., 1.],
#        [1., 1.]], requires_grad=True)

y = x+2
print(y)
#tensor([[3., 3.],
#        [3., 3.]], grad_fn=<AddBackward0>)

z = y*y*3  #平方x3
print(x)
#tensor([[27., 27.],
#        [27., 27.]], grad_fn=<MulBackward0>) 

out = z.mean() #求均值
print(out)
#tensor(27., grad_fn=<MeanBackward0>)

图1.1

想要修改的话:x.requires_grad_(True),就把x变为可以追踪(中间计算结果都保存)的数了。

不想修改了

python 复制代码
with torch.no_gard():
    c = (a * a).sum()  #tensor(151.6830),此时c没有gard_fn

把内容放在torch.no_gard():下就可以了

梯度计算

计算梯度 :直接调用**.backward()**就行。

例如上面图1.1,out.backward()就行

获取梯度 :调用那个数**.gard**

注意:如果.grad在循环中,每次都会和之前的.grad叠加。

所以每次反向传播之后,都需要把梯度变为0再进行操作。


使用.data获取里面的数据。相当于直接拷贝了一份数据,而不改变原数据的内容。

相关推荐
爱昏羔6 小时前
下篇:从检索到交互 — 物流RAG问答系统与WebUI全实现
python·langchain·agent
_Jimmy_6 小时前
Agent引用数据库知识过时的增量同步方案
人工智能·python·langchain
min(a,b)7 小时前
学习第 4 天:面向对象与异常处理
python·学习·学习方法
yangshicong9 小时前
第19章:AI安全防护与AI安全
人工智能·python·安全·prompt·ai编程
果汁华9 小时前
Function Calling 与 Python 实战完整指南
开发语言·网络·python
c_lb72889 小时前
2026年不同基础做量化,先找AI能参与的位置
人工智能·python
chenment11 小时前
ComfyUI 自定义节点开发:从零扩展你的图像生成工作流
python·stable diffusion
IT空门:门主11 小时前
Python 基础语法学习路线图
网络·python·学习
互联网中的一颗神经元12 小时前
小白python入门 - 25. SQL 与表设计入门
数据库·python·sql
RSTJ_162513 小时前
PYTHON+AI LLM DAY ONE HUNDRED AND EIGHTEEN
python