news 2026/4/25 4:16:38

pytorch深度学习笔记

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
pytorch深度学习笔记

目录

摘要

1.神经网络的梯度计算

2.梯度下降法(Gradient Descent)


摘要

本篇文章继续学习尚硅谷深度学习教程,学习内容是神经网络的梯度计算,代码实现用梯度下降法

1.神经网络的梯度计算

在神经网络的学习中,梯度的计算非常重要。神经网络中的梯度,指的就是损失函数关于权重参数的梯度。我们以一个单层的简单网络为例,形状为2×3,权重参数为W,损失函数记为L。那么它的权重参数和梯度为:

这里,梯度也是一个2×3的矩阵,其中各个元素由L关于W中各元素的偏导数构成。

代码如下:

import numpy as np from common.functions import softmax,cross_entropy_error from common.gradient import numerical_gradient class simpleNet: def __init__(self): self.W = np.random.randn(2,3) def forward(self, x): a= x @ self.W return softmax(a) def loss(self, x, t): y = self.forward(x) loss = cross_entropy_error(y, t) return loss x = np.array([0.6, 0.9]) t = np.array([0, 0, 1]) net = simpleNet() f = lambda w: net.loss(x, t) dW = numerical_gradient(f, net.W) print(dW)

测试:

2.梯度下降法(Gradient Descent

梯度下降法(Gradient Descent是一种用于最小化目标函数的迭代优化算法。核心是沿着目标函数(如损失函数)的负梯度方向逐步调整参数,从而逼近函数的最小值。梯度方向指示了函数增长最快的方向,因此负梯度方向是函数下降最快的方向。

具体来说,我们初始找到函数f(x1,x2)的一个点(x1,x2),按下式进行更新

这样就可以沿着负梯度方向,找到一个新的点(x1',x2'),让函数值更小。这里的η表示每次的更新量,在神经网络的学习过程中,就代表了一次学习的步长(一次学习多少、多大程度去更新参数),称为学习率(learning rate)。学习率需要预先设定好,过大或过小都会导致学习效果不佳。

梯度下降法可以代码实现如下:

import numpy as np import matplotlib.pyplot as plt from common.gradient import numerical_gradient def gradient_descent(f, init_x, lr=0.01, step_num=100): x = init_x #保存x变化 x_history = [] for i in range(step_num): x_history.append( x.copy() ) grad = numerical_gradient(f, x) x -= lr * grad return x, np.array(x_history)

案例测试

#定义目标函数f(x)=x1^2+x2^2 def f(x): return x[0]**2 + x[1]**2 if __name__ == '__main__': init_x=np.array([-3.0,4.0]) #超参数 lr=0.1 num_iter=20 #梯度下降法计算最小值 x, x_history = gradient_descent(f, init_x, lr, num_iter) print("最小值点为:",x)

测试

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/21 12:22:21

【完整源码+数据集+部署教程】危险场景检测系统源码分享[一条龙教学YOLOV8标注好的数据集一键训练_70+全套改进创新点发刊_Web前端展示]

一、背景意义 随着城市化进程的加快和工业化水平的提高,危险场景的发生频率逐渐上升,给人们的生命财产安全带来了严重威胁。传统的危险场景监测手段往往依赖于人工巡查和简单的监控设备,存在反应慢、覆盖面窄等缺陷,难以实现实时、…

作者头像 李华
网站建设 2026/4/17 14:03:28

考研加油上岸祝福弹窗程序

https://www.bilibili.com/video/BV1zdBFBbEvj/https://www.bilibili.com/video/BV1zdBFBbEvj/ GraduateAnchor - 考研祝福弹窗程序​ 项目简介 GraduateAnchor(考研上岸)是一个充满温暖与祝福的桌面应用程序,专为考研学子设计。程序运行后…

作者头像 李华
网站建设 2026/4/18 12:54:29

【开题答辩全过程】以 基于Java的打车拼车系统的设计与实现为例,包含答辩的问题和答案

个人简介一名14年经验的资深毕设内行人,语言擅长Java、php、微信小程序、Python、Golang、安卓Android等开发项目包括大数据、深度学习、网站、小程序、安卓、算法。平常会做一些项目定制化开发、代码讲解、答辩教学、文档编写、也懂一些降重方面的技巧。感谢大家的…

作者头像 李华
网站建设 2026/4/18 16:12:44

算法杂谈:回溯路线

目录 前言 在动态规划中: 在bfs中: 前言 对于普通的路线问题,我们可以存储全局变量path存储路线过程中的,一个个“点”。由于这些点就是按照顺序存储的,路线就是可以直接得到的。 但是如果是动态规划,…

作者头像 李华
网站建设 2026/4/18 20:32:49

Langchain-Chatchat如何处理嵌套引用?复杂文档结构解析

Langchain-Chatchat如何处理嵌套引用?复杂文档结构解析 在企业知识库系统日益普及的今天,一个核心挑战浮出水面:如何让AI真正“读懂”那些充满脚注、交叉引用和层级结构的专业文档?比如一份科研报告中写着“详见[1]”,…

作者头像 李华