news 2026/4/15 7:21:03

张量的基础运算

作者头像

张小明

前端开发工程师

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/4/15 7:18:23

2025年12月20日最热门的开源项目(Github)

以下是对本期榜单的综合分析: 1. 总体趋势分析 新老项目并存:榜单中既有2016年、2017年创建的老牌项目(如PayloadsAllTheThings、awesome-mac、every-programmer-should-know),也有2024年及以后创建的全新项目&#x…

作者头像 李华
网站建设 2026/4/15 7:20:31

高效软件测试团队的组织结构与能力建设

在数字化转型加速的2025年,软件质量已成为企业核心竞争力。测试团队作为质量守门人,其组织效能直接决定产品交付速度和用户体验。本文基于现代软件工程实践,深入探讨测试团队的架构设计与能力演进路径。一、测试团队组织结构设计1.1 混合型团…

作者头像 李华
网站建设 2026/4/11 16:32:28

Excalidraw本地化部署教程:私有化环境搭建指南

Excalidraw本地化部署教程:私有化环境搭建指南 在企业级协作日益依赖可视化表达的今天,一张架构图、一个流程草图往往承载着关键决策信息。而将这些敏感内容上传至公网工具——哪怕只是临时分享——都可能带来数据泄露的风险。正是在这种背景下&#xf…

作者头像 李华
网站建设 2026/4/14 7:29:02

Excalidraw与Notion集成:打造一体化知识库

Excalidraw与Notion集成:打造一体化知识库 在技术团队的日常协作中,你是否经历过这样的场景?一场头脑风暴后,白板上画满了系统架构草图和流程逻辑,大家意犹未尽。会议结束,照片被丢进群聊,几天…

作者头像 李华
网站建设 2026/4/8 8:23:30

智能测试用例生成:是银弹还是玩具?面试中如何客观评价?

在2025年的软件测试行业,随着人工智能和机器学习技术的飞速发展,智能测试用例生成工具已成为热门话题。它们被誉为能自动化生成测试数据、场景和路径的“革命性创新”,但同时也被质疑为脱离实际、难以落地的“玩具”。对于软件测试从业者而言…

作者头像 李华
网站建设 2026/4/14 14:50:02

【大模型开发者必看】:Open-AutoGLM电子书高清完整版下载(附源码解析)

第一章:Open-AutoGLM电子书下载 获取《Open-AutoGLM》电子书是深入了解该开源项目自动化推理与代码生成能力的第一步。该项目围绕轻量化大语言模型在代码任务中的应用展开,提供了完整的文档支持与实践案例。 官方发布渠道 电子书可通过项目官网及 GitHu…

作者头像 李华