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/9 9:08:20

张量的运算函数

张量的基本运算函数张量运算在深度学习和科学计算中至关重要。常见的张量运算函数包括加法、减法、乘法和除法。这些运算通常按元素进行,要求参与运算的张量具有相同的形状。加法运算:import torch a torch.tensor([1, 2, 3]) b torch.tensor([4, 5, 6…

作者头像 李华
网站建设 2026/6/8 22:59:26

NeurIPS 2025 Best Paper 一作邱子涵在线开讲!

点击蓝字关注我们AI TIME欢迎每一位AI爱好者的加入!AITIME01NeurIPS 2025 Best Paper AwardAITIME02观看地址A微信视频号直播点击预约AI TIME 视频号直播BBilibili直播进入Bilibili直播间观看,提问有可能会被选中由讲者回答!欢迎关注AITIME论…

作者头像 李华
网站建设 2026/6/8 20:12:58

Excalidraw如何助力创业团队快速迭代产品原型

Excalidraw:创业团队原型迭代的“思维加速器” 在一次典型的创业团队周会上,产品经理刚提出一个新功能设想——“用户可以在任务卡片上标记紧急程度,支持高/中/低三级”。还没等她打开Figma,技术负责人已经打开了浏览器中的Excal…

作者头像 李华
网站建设 2026/6/9 14:41:35

好散文的秘密

力荐,关于写作,这本书不仅讲道,也讲了术。每个人都会汉字,但堆叠起来却又千人千面。什么算好散文,好文章有哪些技巧,这本书会给我们一些答案。如果细数 ai 时代什么不能替代,那么有温度的文字就…

作者头像 李华