0

0

Python PyTorch张量操作_Python怎么用PyTorch进行张量计算和自动求导

冷漠man

冷漠man

发布时间:2025-12-22 11:16:56

|

924人浏览过

|

来源于php中文网

原创

pytorch的核心是张量和自动求导,支持gpu加速与梯度自动计算;需设requires_grad=true开启求导,标量输出调用backward()触发反向传播,梯度存于.grad属性,常用torch.no_grad()、detach()和zero_grad()管理计算图与梯度。

python pytorch张量操作_python怎么用pytorch进行张量计算和自动求导 - php中文网

PyTorch 的核心是 张量(Tensor)自动求导(Autograd),它让深度学习中的数值计算既高效又灵活。你不需要手动推导梯度,只要张量启用了梯度追踪,PyTorch 就能在反向传播时自动计算所有梯度。

创建和操作张量

张量类似于 NumPy 数组,但支持 GPU 加速和自动求导:

  • torch.tensor() 创建: x = torch.tensor([1.0, 2.0, 3.0], requires_grad=True) —— requires_grad=True 是开启自动求导的关键
  • 常用创建方式还有 torch.zeros(3, 4)torch.randn(2, 3)torch.ones_like(x)
  • 支持标准数学运算:y = x ** 2 + 2 * x + 1,结果 y 也会继承 requires_grad=True(只要输入有梯度)
  • 注意:原地操作(如 x.add_(1))可能破坏计算图,一般避免在需求导的张量上使用带下划线的就地方法

执行自动求导

自动求导只对**标量输出**调用 .backward() 有效(即最终损失是一个数):

  • y 是标量(如 loss),直接调用 y.backward(),PyTorch 会从 y 开始反向遍历计算图,把梯度存入各叶子张量的 .grad 属性中
  • 例如:x = torch.tensor(2.0, requires_grad=True); y = x**3; y.backward(); print(x.grad) 输出 tensor(12.)(因为 dy/dx = 3x² = 12)
  • 若输出不是标量(比如向量),需传入与输出同形状的 gradient 参数,例如 y.backward(torch.ones_like(y)),等价于对每个元素求和后再反向

管理计算图和梯度

PyTorch 默认每次前向传播都会构建新计算图,反向传播后图自动释放。常见操作包括:

Sora
Sora

Sora是OpenAI发布的一种文生视频AI大模型,可以根据文本指令创建现实和富有想象力的场景。

下载

立即学习Python免费学习笔记(深入)”;

  • with torch.no_grad(): 块内所有张量操作不记录梯度,用于推理、模型评估或参数更新时防止干扰
  • x.detach() 返回一个与 x 数据相同但无梯度历史的新张量(常用于从计算图中“切出”一部分)
  • 清空梯度:训练循环中通常要写 optimizer.zero_grad() 或手动 model.zero_grad(),等价于对所有可训练参数的 .grad 置零
  • 想重复对同一张量调用 backward()?默认会累加梯度,首次后需手动置零,或传参 retain_graph=True 保留图(慎用,占内存)

实战小例子:手动实现线性回归梯度更新

理解原理最直接的方式是写一小段可运行代码:

  • 定义数据:x = torch.randn(100, 1); y_true = 2 * x + 1 + 0.1 * torch.randn(100, 1)
  • 初始化参数:w = torch.randn(1, 1, requires_grad=True); b = torch.randn(1, requires_grad=True)
  • 前向:y_pred = x @ w + b;损失:loss = ((y_pred - y_true) ** 2).mean()
  • 反向:loss.backward();更新:w.data -= 0.01 * w.grad; b.data -= 0.01 * b.grad(注意用 .datatorch.no_grad() 避免更新进入图)

掌握张量创建、运算规则、requires_grad 开关、backward() 触发条件和梯度管理,就掌握了 PyTorch 自动求导的主干。它不复杂,但细节决定是否能稳定训练模型。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
python中print函数的用法
python中print函数的用法

python中print函数的语法是“print(value1, value2, ..., sep=' ', end=' ', file=sys.stdout, flush=False)”。本专题为大家提供print相关的文章、下载、课程内容,供大家免费下载体验。

192

2023.09.27

python print用法与作用
python print用法与作用

本专题整合了python print的用法、作用、函数功能相关内容,阅读专题下面的文章了解更多详细教程。

17

2026.02.03

pytorch是干嘛的
pytorch是干嘛的

pytorch是一个基于python的深度学习框架,提供以下主要功能:动态图计算,提供灵活性。强大的张量操作,实现高效处理。自动微分,简化梯度计算。预构建的神经网络模块,简化模型构建。各种优化器,用于性能优化。想了解更多pytorch的相关内容,可以阅读本专题下面的文章。

457

2024.05.29

Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习

PyTorch 是一种用于构建深度学习模型的功能完备框架,是一种通常用于图像识别和语言处理等应用程序的机器学习。 使用Python 编写,因此对于大多数机器学习开发者而言,学习和使用起来相对简单。 PyTorch 的独特之处在于,它完全支持GPU,并且使用反向模式自动微分技术,因此可以动态修改计算图形。

27

2025.12.22

Golang 测试体系与代码质量保障:工程级可靠性建设
Golang 测试体系与代码质量保障:工程级可靠性建设

Go语言测试体系与代码质量保障聚焦于构建工程级可靠性系统。本专题深入解析Go的测试工具链(如go test)、单元测试、集成测试及端到端测试实践,结合代码覆盖率分析、静态代码扫描(如go vet)和动态分析工具,建立全链路质量监控机制。通过自动化测试框架、持续集成(CI)流水线配置及代码审查规范,实现测试用例管理、缺陷追踪与质量门禁控制,确保代码健壮性与可维护性,为高可靠性工程系统提供质量保障。

0

2026.02.28

Golang 工程化架构设计:可维护与可演进系统构建
Golang 工程化架构设计:可维护与可演进系统构建

Go语言工程化架构设计专注于构建高可维护性、可演进的企业级系统。本专题深入探讨Go项目的目录结构设计、模块划分、依赖管理等核心架构原则,涵盖微服务架构、领域驱动设计(DDD)在Go中的实践应用。通过实战案例解析接口抽象、错误处理、配置管理、日志监控等关键工程化技术,帮助开发者掌握构建稳定、可扩展Go应用的最佳实践方法。

2

2026.02.28

Golang 性能分析与运行时机制:构建高性能程序
Golang 性能分析与运行时机制:构建高性能程序

Go语言以其高效的并发模型和优异的性能表现广泛应用于高并发、高性能场景。其运行时机制包括 Goroutine 调度、内存管理、垃圾回收等方面,深入理解这些机制有助于编写更高效稳定的程序。本专题将系统讲解 Golang 的性能分析工具使用、常见性能瓶颈定位及优化策略,并结合实际案例剖析 Go 程序的运行时行为,帮助开发者掌握构建高性能应用的关键技能。

1

2026.02.28

Golang 并发编程模型与工程实践:从语言特性到系统性能
Golang 并发编程模型与工程实践:从语言特性到系统性能

本专题系统讲解 Golang 并发编程模型,从语言级特性出发,深入理解 goroutine、channel 与调度机制。结合工程实践,分析并发设计模式、性能瓶颈与资源控制策略,帮助将并发能力有效转化为稳定、可扩展的系统性能优势。

13

2026.02.27

Golang 高级特性与最佳实践:提升代码艺术
Golang 高级特性与最佳实践:提升代码艺术

本专题深入剖析 Golang 的高级特性与工程级最佳实践,涵盖并发模型、内存管理、接口设计与错误处理策略。通过真实场景与代码对比,引导从“可运行”走向“高质量”,帮助构建高性能、可扩展、易维护的优雅 Go 代码体系。

16

2026.02.27

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 4.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.7万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号