news 2026/6/10 16:23:37

张量的基础运算

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
张量的基础运算

张量的基本概念

张量是多维数组的泛化概念,标量(0维)、向量(1维)、矩阵(2维)均可视为张量的特例。在深度学习和科学计算中,张量是核心数据结构,支持高效的数值运算。

创建张量

以Python库numpytorch为例:

import numpy as np import torch # 创建numpy数组(张量) np_tensor = np.array([[1, 2], [3, 4]]) # 创建PyTorch张量 torch_tensor = torch.tensor([[1, 2], [3, 4]])

算术运算

张量支持逐元素运算(加减乘除)和广播机制:

a = torch.tensor([1, 2]) b = torch.tensor([3, 4]) # 逐元素加法 result_add = a + b # 输出: tensor([4, 6]) # 逐元素乘法 result_mul = a * b # 输出: tensor([3, 8])

矩阵乘法

使用@torch.matmul进行矩阵乘法:

A = torch.tensor([[1, 2], [3, 4]]) B = torch.tensor([[5, 6], [7, 8]]) # 矩阵乘法 result_matmul = A @ B # 输出: tensor([[19, 22], [43, 50]])

张量变形

通过reshapeview改变张量形状:

x = torch.arange(6) # tensor([0, 1, 2, 3, 4, 5]) y = x.reshape(2, 3) # tensor([[0, 1, 2], [3, 4, 5]])

聚合操作

对张量进行求和、均值等操作:

z = torch.tensor([[1.0, 2.0], [3.0, 4.0]]) # 全局求和 sum_all = z.sum() # 输出: tensor(10.) # 沿维度求和 sum_dim0 = z.sum(dim=0) # 输出: tensor([4., 6.])

梯度计算(自动微分)

PyTorch张量可通过requires_grad=True启用梯度追踪:

x = torch.tensor(2.0, requires_grad=True) y = x ** 2 y.backward() # 计算梯度 print(x.grad) # 输出: tensor(4.0)

张量设备转换

在CPU与GPU之间移动张量:

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

Excalidraw支持多人光标追踪的实时协作体验

Excalidraw 中的实时协作:当光标成为对话的一部分 在一场跨时区的系统设计评审会上,团队成员分布在三个国家。没有冗长的 PPT 汇报,也没有模糊不清的“大概在这里”的口头描述——取而代之的是,每个人都能看到主讲人光标精准地划…

作者头像 李华
网站建设 2026/6/9 22:13:58

打造你的企业级智能文档问答系统——Everything plus RAG 实战指南

从零开始,用开源技术搭建一个能"读懂"文档的智能问答系统大家好,我是铭毅天下。今天给大家分享一个非常实用的项目——Everything plus RAG 智能文档问答系统。相信大家的电脑上都安装了 Everything,一个磁盘文件快速查找桌面级应用软件。一直…

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

Excalidraw与Jira联动:需求评审可视化辅助

Excalidraw与Jira联动:需求评审可视化辅助 在一次典型的需求评审会上,产品经理正试图用一段文字描述一个复杂的审批流程:“当用户提交申请后,系统会根据角色判断是否需要一级审核,如果是外部客户还需触发风控检查……”…

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

vue基于Django框架的面向IT行业的招聘求职推荐系统_a4av7

文章目录系统截图项目简介大数据系统开发流程主要运用技术介绍爬虫核心代码展示结论源码文档获取定制开发/同行可拿货,招校园代理 :文章底部获取博主联系方式!系统截图 vue基于Django框架的面向IT行业的招聘求职推荐系统_a4av7 项目简介 本次…

作者头像 李华
网站建设 2026/6/10 14:24:31

Excalidraw性能优化:处理大型图表的流畅体验

Excalidraw性能优化:处理大型图表的流畅体验 在当今远程协作日益频繁的背景下,技术团队对可视化工具的需求早已超越了简单的“画图”。一张架构图可能承载着整个系统的演进脉络,一个流程图或许就是产品逻辑的核心表达。而当这些图形变得越来越…

作者头像 李华