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)
相关推荐
阿里云大数据AI技术9 小时前
云栖实录 | 实时计算 Flink 全新升级 - 全栈流处理平台助力实时智能
大数据·人工智能
新加坡内哥谈技术9 小时前
谷歌Quantum Echoes算法:迈向量子计算现实应用的重要一步
人工智能
分布式存储与RustFS10 小时前
RustFS:MinIO的“平替”还是“乱杀”?
python·rust·对象存储·minio·存储·企业存储·rustfs
爱幻想-hjyp10 小时前
【Python】uv包管理器常用命令
开发语言·python·uv
用什么都重名10 小时前
UV工具安装配置使用教程
python·uv·包管理器
Godspeed Zhao10 小时前
自动驾驶中的传感器技术70——Navigation(7)
人工智能·机器学习·自动驾驶
Python×CATIA工业智造10 小时前
Python装饰器解包装技术详解:从原理到高级应用
python·pycharm
这儿有一堆花10 小时前
AI 翻译入门指南:机器如何理解语言
人工智能·web
明月照山海-10 小时前
机器学习周报十九
人工智能·机器学习
王哈哈^_^10 小时前
PyTorch vs TensorFlow:从入门到落地的全方位对比
人工智能·pytorch·python·深度学习·计算机视觉·tensorflow·1024程序员节