news 2026/4/22 21:35:34

张量的基础运算

作者头像

张小明

前端开发工程师

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/18 17:37:33

刚刚,AI 大神 Karpathy 2025 年度总结刷屏!

本文来自 Andrej Karpathyhttps://karpathy.bearblog.dev/year-in-review-2025/转自:赛博禅心2025 年是大语言模型突飞猛进的一年以下是我个人认为最值得关注的几个「范式转变」,这些变化重塑了整个行业格局,也在概念上给我留下了深刻印象可验…

作者头像 李华
网站建设 2026/4/18 3:24:40

放下等。。。就好了的心态

依般若波罗蜜多故,心无挂碍,无挂碍故,无有恐怖,远离颠倒梦想,究竟涅槃。“放下‘等……就好了’的心态”,恰恰是 “4 me理论”在实践层面最关键、最深刻的落地法则,是“为我”原则对治一种普遍存…

作者头像 李华
网站建设 2026/4/23 1:41:35

某程序员爆料:没买房被组长针对了,他自己的房估计亏了五百万,现在对我总阴阳怪气,说我不买房干嘛,还说我的存款就算~

来自:网络,侵删推荐一个程序员编程资料站:http://cxyroad.com副业赚钱专栏:https://xbt100.top2024年IDEA最新激活方法后台回复:激活码CSDN免登录复制代码插件下载:CSDN复制插件以下是正文。刚看到个贴子&a…

作者头像 李华
网站建设 2026/4/22 22:33:30

Open-AutoGLM穿衣推荐系统(90%准确率背后的模型秘密)

第一章:Open-AutoGLM穿衣推荐系统(90%准确率背后的模型秘密)Open-AutoGLM 是一款基于多模态大语言模型的智能穿衣推荐系统,融合了视觉理解、气候感知与用户偏好建模,在真实场景中实现了高达90%的推荐准确率。其核心在于…

作者头像 李华
网站建设 2026/4/19 21:00:01

【AI日程管理新突破】:基于Open-AutoGLM的生日提醒系统设计全公开

第一章:AI日程管理新突破概述人工智能技术正以前所未有的速度重塑个人与组织的时间管理方式。在日程规划领域,新一代AI系统通过自然语言理解、上下文感知和预测性分析,实现了从被动记录到主动协调的跨越。这些智能助手不仅能解析模糊指令&…

作者头像 李华