news 2026/6/10 18:20:44

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/6/10 10:18:25

如何配置华为云国际站代理商OBS的跨区域复制?

配置华为云国际站代理商 OBS 跨区域复制(CRR),核心是完成 “前置准备 IAM 委托 规则配置 验证监控” 四步,代理商可全程协助账号 / 配额 / 合规与成本优化,确保跨境数据异步复制稳定、安全且成本可控。以下是可直接…

作者头像 李华
网站建设 2026/6/10 13:51:57

Langchain-Chatchat上下文窗口优化技巧

Langchain-Chatchat 上下文窗口优化实践:如何在有限 token 中榨出最大知识价值 在企业级智能问答系统中,一个看似不起眼的数字常常成为决定成败的关键——上下文长度。8192?32768?这些冷冰冰的 token 数字背后,是模型…

作者头像 李华
网站建设 2026/6/10 16:50:53

Langchain-Chatchat向量数据库选型建议(Chroma/FAISS/Milvus)

Langchain-Chatchat向量数据库选型建议(Chroma/FAISS/Milvus) 在构建本地知识库问答系统时,一个常见的挑战是:如何让大语言模型(LLM)准确回答基于企业私有文档的问题?毕竟,通用模型并…

作者头像 李华
网站建设 2026/6/10 16:57:15

智能体之构建长短期记忆:深入解析 mem0 框架与实战

摘要:大模型(LLM)天生是无状态的,但在构建真正可用的 AI Agent(智能体)时,记忆能力是区分“玩具”与“产品”的关键分水岭。本文将深入探讨智能体长短期记忆的设计哲学,引入下一代记…

作者头像 李华