Pytorch基础入门4

目录

4.3:求最小值

下面讲一个小案例:使用自动微分机制配套使用SGD随机梯度下降来求函数最小值。

python 复制代码
import torch 

x = torch.tensor(0.0,requires_grad=True)
a = torch.tensor(1.0)
b = torch.tensor(-2.0)
c= torch.tensor(1.0)

optimizer = torch.optim.SGD(params=[x],lr=0.01)
print(optimizer)

def f(x):
    result = a*torch.pow(x,2) +b*x +c 
    return(result)

for i in range(500):
    optimizer.zero_grad()
    y=f(x)
    y.backward()
    optimizer.step()
print("y=", y.data, ";", "x=", x.data)    

结果如下:

powershell 复制代码
SGD (
Parameter Group 0
    dampening: 0
    differentiable: False
    foreach: None
    fused: None
    lr: 0.01
    maximize: False
    momentum: 0
    nesterov: False
    weight_decay: 0
)
y= tensor(0.) ; x= tensor(1.0000)
相关推荐
Dxy123931021613 分钟前
python如何通过链接下载保存视频
python·spring·音视频
Terio_my42 分钟前
Java bean 数据校验
java·开发语言·python
Elastic 中国社区官方博客1 小时前
Elasticsearch MCP 服务器:与你的 Index 聊天
大数据·服务器·人工智能·elasticsearch·搜索引擎·ai·全文检索
无咎.lsy2 小时前
裸K初级篇 - (一)蜡烛突破信号
python
virtual_k1smet3 小时前
#等价于e * d ≡ 1 mod φ(n) #模逆元详解
人工智能·算法·机器学习
可触的未来,发芽的智生3 小时前
新奇特:神经网络的集团作战思维,权重共享层的智慧
人工智能·python·神经网络·算法·架构
H3C-Navigator3 小时前
HRPC在Polaris存储系统中的应用
网络·人工智能·ai-native
智慧地球(AI·Earth)4 小时前
智能体版中科院学术GPT上线内测!AI与科研的深度碰撞
人工智能·gpt·科研助手·学术智能体
文火冰糖的硅基工坊4 小时前
[创业之路-653]:社会产品与服务的分类
大数据·数据库·人工智能
jerryinwuhan4 小时前
Python数据挖掘之基础分类模型_支持向量机(SVM)
python·支持向量机·数据挖掘