news 2026/8/7 23:42:26

张量的运算函数

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
张量的运算函数

张量的基本运算函数

张量运算在深度学习和科学计算中至关重要。常见的张量运算函数包括加法、减法、乘法和除法。这些运算通常按元素进行,要求参与运算的张量具有相同的形状。

加法运算:

import torch a = torch.tensor([1, 2, 3]) b = torch.tensor([4, 5, 6]) c = a + b # 结果为 tensor([5, 7, 9])

乘法运算:

d = a * b # 结果为 tensor([4, 10, 18])

矩阵乘法运算

矩阵乘法是深度学习中的核心操作,使用matmul@运算符实现。矩阵乘法要求第一个张量的最后一维与第二个张量的倒数第二维匹配。

矩阵乘法示例:

x = torch.randn(3, 4) y = torch.randn(4, 5) z = torch.matmul(x, y) # 结果为 3x5 张量

批量矩阵乘法:

batch_x = torch.randn(10, 3, 4) batch_y = torch.randn(10, 4, 5) batch_z = torch.bmm(batch_x, batch_y) # 结果为 10x3x5 张量

张量缩减操作

缩减操作沿着指定维度对张量进行汇总计算,包括求和、求均值、最大值和最小值等。

求和运算:

t = torch.tensor([[1, 2], [3, 4]]) sum_all = torch.sum(t) # 结果为 10 sum_dim0 = torch.sum(t, dim=0) # 结果为 tensor([4, 6])

最大值运算:

max_val, max_idx = torch.max(t, dim=1) # 返回值和索引

张量变形操作

变形操作改变张量的形状而不改变其数据,包括viewreshapetranspose等。

改变形状:

t = torch.arange(6) t_reshaped = t.view(2, 3) # 变为 2x3 张量

转置操作:

t = torch.tensor([[1, 2], [3, 4]]) t_transposed = t.t() # 变为 2x2 转置矩阵

广播机制

广播机制允许不同形状的张量进行运算,较小的张量会自动扩展以匹配较大张量的形状。

广播示例:

a = torch.tensor([1, 2, 3]) b = torch.tensor([[0], [1], [2]]) c = a + b # b 广播为 3x3 张量

高级索引操作

高级索引允许复杂的数据选择和修改,包括布尔索引和整数数组索引。

布尔索引:

t = torch.tensor([1, 2, 3, 4]) mask = t > 2 selected = t[mask] # 结果为 tensor([3, 4])

整数数组索引:

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

Excalidraw离线使用方案:PWA应用部署指南

Excalidraw离线使用方案:PWA应用部署指南 在航班爬升、地铁穿行或内网隔离的会议室里,你是否曾因网络中断而被迫中断正在绘制的系统架构图?对于依赖可视化工具进行快速表达的技术人员而言,这种“断联即失能”的体验尤为致命。Exca…

作者头像 李华
网站建设 2026/8/7 21:36:41

Excalidraw移动端使用体验评测与改进建议

Excalidraw移动端使用体验与优化方向 在远程协作日益成为常态的今天,如何快速、直观地表达复杂的技术构想,已成为工程师、产品经理和设计师共同面对的挑战。传统的流程图工具往往显得过于规整冰冷,而手绘草图虽有亲和力却难以共享与迭代。Exc…

作者头像 李华
网站建设 2026/8/7 2:59:23

基于Django的智慧旅游系统行程分享的vue 爬虫 可视化大屏

文章目录系统截图项目简介大数据系统开发流程主要运用技术介绍爬虫核心代码展示结论源码文档获取定制开发/同行可拿货,招校园代理 :文章底部获取博主联系方式!系统截图 基于Django的智慧旅游系统行程分享的vue 爬虫 可视化大屏 项目简介 本次研究…

作者头像 李华
网站建设 2026/8/7 8:20:16

搭建单相光伏MPPT并网模型:基于2018b版本的探索与实践

单相光伏mppt并网模型 根据网上视频搭建的,可以跟着学,内有一些自己的理解注释。 2018b最近在研究光伏相关的内容,成功搭建了一个单相光伏MPPT并网模型,这个模型是参照网上视频一步步构建起来的,过程中加入了不少自己的…

作者头像 李华
网站建设 2026/8/7 19:05:40

4 me 理论与受苦、创造、体验三层意义

将“受苦、创造、体验”整合为“4 me理论”的三层意义结构,是一次极为深刻的哲学深化。这使该理论从一个自我认知模型,升级为一个能解释存在核心活动、并指引如何赋予生命以意义的价值论框架。这“三层意义”并非平行,而是一个从被动接受到主…

作者头像 李华
网站建设 2026/8/7 7:20:23

计算机新书销量 TOP1

这本书不需要虚构的赞美。它是 BookAuthority 算法评选出的 R 书单第 1 名;它在 Google Scholar 上被学术界引用了近 2000 次;它是哈佛、斯坦福等顶尖高校数据科学课程的指定教材;它是 R 社区公认的“圣经”。就像学物理绕不开牛顿定律一样&a…

作者头像 李华