news 2026/2/7 12:36:23

张量的基础运算

作者头像

张小明

前端开发工程师

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/2/3 20:31:29

基于大数据的专业智能导学系统的设计与实现-计算机毕业设计源码+LW文档

摘要 随着新世纪无纸化办公方式的普及,自动化信息处理和基于网络的信息交互方式已被广泛应用。现在很多行业基本上都是交由计算机进行管理和测试,网络与计算机已成为整个线上管理体系中的重要组成部分。虽然信息技术广泛应用和数据存取更加方便&#xff…

作者头像 李华
网站建设 2026/2/3 12:51:14

【Open-AutoGLM玩家必备】:6款高性价比礼物推荐,内行人都在悄悄买

第一章:Open-AutoGLM礼物选购推荐在智能AI时代,个性化礼物正逐渐成为表达心意的新方式。Open-AutoGLM作为一款基于开源大模型驱动的智能推荐系统,能够根据用户画像、兴趣标签和预算范围,自动生成精准的礼物推荐方案。无论是生日、…

作者头像 李华
网站建设 2026/2/5 9:56:58

测试AIGC应用:当输出不再是确定性结果

测试范式的根本性转变 随着生成式人工智能(AIGC)技术在各行业的深入应用,软件测试领域正面临前所未有的挑战。传统的二进制断言(True/False)测试框架在应对非确定性、创造性输出的AIGC系统时显露出局限性,…

作者头像 李华
网站建设 2026/2/6 6:01:59

还在手动记生日?,用Open-AutoGLM实现全自动节日提醒

第一章:还在手动记生日?告别遗忘的烦恼你是否经常因为忘记朋友或家人的生日而感到尴尬?在快节奏的现代生活中,依赖记忆已不再可靠。借助自动化工具和简单的代码脚本,我们可以轻松实现生日提醒系统,彻底告别…

作者头像 李华
网站建设 2026/2/3 18:06:32

为什么顶尖科技公司都在用Open-AutoGLM做资讯聚合?

第一章:Open-AutoGLM 新闻资讯聚合Open-AutoGLM 是一个基于开源大语言模型的智能新闻聚合系统,专注于自动化采集、语义理解与个性化推荐。该系统融合了网络爬虫、自然语言处理与用户行为分析技术,能够实时从多个新闻源中提取高质量内容&#…

作者头像 李华