当前位置：首页 > news >正文

20250110_ PyTorch中的张量操作

news 来源：原创 2025/5/13 3:19:28

文章目录

前言
1、torch.cat 函数
2、索引、维度扩展和张量的广播
3、切片操作
- 3.1、 encoded_first_node
- 3.2、probs
4、长难代码分析
- 4.1、selected
- - 4.1.1、multinomial(1)工作原理：
总结

前言

1、torch.cat 函数

torch.cat 函数将两个张量拼接起来，具体地是在第三个维度（dim=2）上进行拼接。注：dim取值范围是0~2

node_xy_demand = torch.cat((node_xy, node_demand[:, :, None]), dim=2)

其中所用参数为：

node_xy = reset_state.node_xy
# shape: (batch, problem, 2)
node_demand = reset_state.node_demand
# shape: (batch, problem)

若要拼接node_xy 与node_demand 需要将node_demand 进行维度拓展即 node_demand[:, :, None])

node_xy = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

node_demand = torch.tensor([[[10], [20]], [[30], [40]]])

node_xy_demand = torch.tensor([[[ 1,  2, 10], [ 3,  4, 20]],[[ 5,  6, 30], [ 7,  8, 40]]])

2、索引、维度扩展和张量的广播

_ = self.decoder.regret_embedding[None, None, :].expand(encoded_nodes.size(0), 1, self.decoder.regret_embedding.size(-1))

self.decoder.regret_embedding是一个张量。
self.decoder.regret_embedding[None, None, :]增加regret_embedding的维度。维度扩展成 (1, 1, D)

.expand(encoded_nodes.size(0), 1, self.decoder.regret_embedding.size(-1))

expand 用来沿特定维度复制张量，以实现广播。
encoded_nodes.size(0) 返回的是 encoded_nodes 张量的第一个维度大小。
1 表示第二个维度的大小。
self.decoder.regret_embedding.size(-1) 返回的是 self.decoder.regret_embedding 的最后一个维度的大小，也就是嵌入的维度 D

总结：将张量建立为所需维度在此为三维，使用expand沿着新建维度进行拓展到所需形状。

3、切片操作

3.1、 encoded_first_node

 encoded_first_node = self.encoded_nodes[:, [0], :]

这行代码中的切片操作是从 self.encoded_nodes 中提取特定的数据部分：

: 表示选择所有批次的样本，保留第一个维度（batch）。
[0] 表示选择每个样本中的第一个节点，因此提取的是第一个节点的嵌入向量。
: 表示选择该节点的所有嵌入维度，即保留第三个维度（embedding）的所有值。

最终，经过这些操作，encoded_first_node 的形状为 (batch, 1, embedding)，即每个样本只包含第一个节点的嵌入向量，保留了嵌入维度。

3.2、probs

probs[:, :, :-1]

这是对 probs 张量的切片操作，作用是从 probs 的第三个维度（即最后一个维度）中移除最后一列。

selected = probs.argmax(dim=2)

argmax(dim=2) 表示在 probs 张量的第3维度（类别维度）上，找到每个样本中概率最大的类别索引。
argmax 返回的是最大值的索引，而不是最大值本身。

4、长难代码分析

4.1、selected

selected = probs.reshape(batch_size * pomo_size, -1).multinomial(1).squeeze(dim=1).reshape(batch_size, pomo_size)

prob的shape: (batch, pomo, problem+1)

probs.reshape(batch_size * pomo_size, -1)：
- 这一步将 probs 的形状从 (batch, pomo, problem + 1) 转变为 (batch * pomo, problem + 1)。
- -1：表示自动推算出第二维的大小（即 problem + 1）
- 新的形状 (batch * pomo, problem + 1)。
multinomial(1)：
- multinomial(1) 用于从给定的概率分布中选择一个类别。它会返回一个形状为 (batch_size * pomo_size, 1) 的张量，每一行选择一个元素的索引，代表从 probs 中选择的元素。
.squeeze(dim=1)：
- squeeze(dim=1) 是去除第二个维度（索引维度），将形状变为 (batch_size * pomo_size)。
.reshape(batch_size, pomo_size)：
- 最后，通过 reshape(batch_size, pomo_size) 将张量恢复到原来的形状 (batch_size, pomo_size)，即每个批次对应一个选择的元素索引。

4.1.1、multinomial(1)工作原理：

输入：
multinomial(1) 需要一个形状为 (N, C) 的张量，其中 N 是样本的数量，C 是类别的数量。这个张量表示每个样本在各个类别下的概率分布。
输出：
multinomial(1) 返回一个形状为 (N, 1) 的张量，每个元素是该样本选择的类别的索引。

具体来说，multinomial(1) 会根据每个类别的概率，从概率分布中选取一个类别。这个选择是随机的，但是会遵循给定的概率分布，即概率较大的类别被选中的几率较高，概率较小的类别被选中的几率较低。

总结

20250110_ PyTorch中的张量操作

文章目录前言1、torch.cat 函数2、索引、维度扩展和张量的广播3、切片操作3.1、 encoded_first_node3.2、probs 4、长难代码分析4.1、selected4.1.1、multinomial(1)工作原理： 总结前言 1、torch.cat 函数 torch.cat 函数将两个张量拼接起来，具体地是…...

编程日记 2025/5/13 3:19:28

【ROS2】☆ launch之Python

☆重点 ROS1和ROS2其中一个很大区别之一就是launch的编写方式。在ROS1中采用xml格式编写launch，而ROS2保留了XML 格式launch，还另外引入了Python和YAML 编写方式。选择哪种编写取决于每位开发人员的爱好，但是ROS2官方推荐使用Python方式编写…...

编程日记 2025/5/9 14:11:01

unity rb.velocity和transform.position

rb.velocity和transform.position是用来控制物体位置的两种方式，前者通常用来控制人物的移动，它们的主要区别和适用场景如下一，rb.velocity（控制刚体的速度） 它可以直接控制物体的速度，而不是物体的位置…...

编程日记 2025/5/6 12:43:33

景芯SOC设计实战

终身辅导、一对一辅导，手把手教您完成SoC全流程设计，从入门到进阶，带您掌握SoC芯片架构、算法、设计、验证、DFT、后端及低功耗全流程！直播视频不定期升级！让您快速超越同龄人！ 景芯团队主打文档服务器实战…...

编程日记 2025/5/9 15:26:29

【WRF运行报错】总结WRF运行时报错及解决方案（持续更新）

目录 ./real.exe错误1：ERROR while reading namelist physics./wrf.exe错误1：FATAL CALLED FROM FILE: <stdin> LINE: 2419 Warning: too many input landuse types参考./real.exe 错误1：ERROR while reading namelist physics 执行./real.exe时，报错如下： taski…...

编程日记 2025/5/6 6:33:40

Mysql快速列出来所有列信息

文章目录需求描述实现思路1、如何查表信息2、如何取字段描述信息3、如何将列信息一行展示4、拼接最终结果需求描述如何将MySQL数据库中指定表【tb_order】的所有字段都展示出来，以备注中的中文名为列名。实现思路最终展示效果，即拼接出可执行执行…...

编程日记 2025/5/9 18:14:36

spring boot发送邮箱，java实现邮箱发送（邮件带附件）3中方式【保姆级教程一，代码直接用】

文章目录 Java发送邮箱的方式1. 基于 Javax.mail 实现关于附件上传的方法 2. 基于 org.apache.commons.mail 实现常见报错 3. 基于 spring-boot-starter-mail 实现（推荐） 实际开发时需要实现邮件发送，本文章实现如何从零实现邮件发送。也就是…...

编程日记 2025/5/11 5:50:08

数据集-目标检测系列- 电话测数据集 call_phone ＞＞ DataBall

数据集-目标检测系列- 电话测数据集 call DataBall 助力快速掌握数据集的信息和使用方式，会员享有百种数据集，持续增加中。需要更多数据资源和技术解决方案，知识星球： “DataBall - X 数据球(free)” 贵在坚持！ …...

编程日记 2025/5/9 5:20:34

Zstandard压缩算法

简介 Zstandard（缩写为zstd）是一种开源的无损数据压缩算法，主要设计目标是提供高比率的压缩和快速的解压缩速度。它由Yann Collet开发，并于2015年首次发布。特点高比率的压缩（通常比gzip更好）。快速的解压缩速度（通常比gzip更快）。支持流式解压缩。可以选择不同的压…...

编程日记 2025/5/6 20:22:24

npm i 报错

nodejs中使用npm install命令时报错 npm err! file C: \user\admin\package.json_package.json 里缺少 description 和 repository 两个n字段。-CSDN博客...

编程日记 2025/4/24 4:34:09

【LeetCode】力扣刷题热题100道（26-30题）附源码轮转数组乘积矩阵螺旋矩阵旋转图像（C++）

目录 1.轮转数组 2.除自身以外数组的乘积 3.矩阵置零 4.螺旋矩阵 5.旋转图像 1.轮转数组给定一个整数数组 nums，将数组中的元素向右轮转 k 个位置，其中 k 是非负数。 class Solution { public:void rotate(vector<int>& nums, int k) …...

编程日记 2025/5/6 22:28:06

EFCore HasDefaultValueSql

今天小伙伴在代码中遇到了有关 HasDefaultValue 的疑问，这里整理澄清下... 在使用 Entity Framework Core (EFCore) 配置实体时，HasDefaultValue 方法会为数据库列设置一个默认值。该默认值的行为取决于以下条件： 1. 配置 HasDefaultValue 的…...

编程日记 2025/4/29 9:53:08

【数据结构】栈

目录 1.1 什么是栈 1.2 顺序栈 1.2.1 特性 1.3 链式栈 1.3.1 特性总结： 1.1 什么是栈栈是只能在一端进行插入和删除操作的线性表(又称为堆栈)，进行插入和删除操作的一端称为栈顶，另一端称为栈底。特点：栈是先进后出FILO…...

编程日记 2025/5/11 13:37:18

C++初阶—CC++内存管理

第一章：C/C内存分布 int globalVar 1; static int staticGlobalVar 1; void Test() {static int staticVar 1;int localVar 1;int num1[10] { 1, 2, 3, 4 };char char2[] "abcd";const char* pChar3 "abcd";int* ptr1 (int*)malloc(si…...

编程日记 2025/5/13 3:15:14

【机器视觉】OpenCV 图像基本变换

文章目录介绍机器视觉的核心组成部分机器视觉的关键技术和趋势 4. 图像的基本变换4.1 图像的放大与缩小4.2 图像的翻转4.3 图像的旋转4.4 仿射变换之图像平移4.5 仿射变换之获取变换矩阵4.6 透视变换介绍机器视觉（Machine Vision）是一门跨学科的领域…...

编程日记 2025/5/10 1:50:31

【数据库】四、数据库管理与维护

文章目录四、数据库管理与维护1 安全性管理2 事务概述3 并发控制4 备份与恢复管理四、数据库管理与维护 1 安全性管理安全性管理是指保护数据库，以避免非法用户进行窃取数据、篡改数据、删除数据和破坏数据库结构等操作三个级别认证： 服务器级别…...

编程日记 2025/5/10 8:47:53

徐克版射雕唤醒热血武侠魂，共赴新春侠义之约

2025年大年初一，由徐克执导的古装武侠电影《射雕英雄传：侠之大者》将在影院拉开帷幕，在精彩纷呈的春节档电影中，“大IP”“大导演”“大场面”等标签让这部电影自定档起便备受关注，其精良的制作和传统中国武侠风的设定…...

编程日记 2025/5/7 5:24:00

设计模式（观察者模式）

设计模式（观察者模式） 第三章设计模式之观察者模式观察者模式介绍观察者模式（Observer Design Pattern） 也被称为发布订阅模式。模式定义：在对象之间定义一个一对多的依赖，当一个对象状态改变的时候…...

编程日记 2025/5/10 22:17:18

能量函数和能量守恒

在之前的文章1中讨论了与循环坐标相对应的动量守恒定律和动量矩守恒定律，本文将由拉格朗日方程中导出能量函数，进一步讨论能量守恒定律，并给出耗散系统的处理方法，这其中用到的一个关键数学定理是欧拉定理（描述如何将一…...

编程日记 2025/5/12 17:43:37

【pycharm发现找不到python打包工具,且无法下载】

发现找不到python打包工具,且无法下载解决方法： 第一步：安装distutils，在CMD命令行输入： python -m ensurepip --default-pip第二步：检查和安装setuptools和wheel： python -m pip install --upgrade …...

编程日记 2025/5/13 2:01:13

使用 Maxwell 计算母线的电动势

三相短路事件的动力学三相短路事件在电气系统中至关重要，因为三相之间的意外连接会导致电流大幅激增。如果管理不当，这些事件可能会造成损坏，因为它们会对电气元件（尤其是母线）产生极大的力和热效应。短路时&#x…...

编程日记 2025/5/10 20:28:36

【Python】Python之Selenium基础教程+实战demo：提升你的测试+测试数据构造的效率！

这里写目录标题什么是Selenium？Selenium基础用法详解环境搭建编写第一个Selenium脚本解析脚本常用的元素定位方法常用的WebDriver方法等待机制 Selenium高级技巧详解页面元素操作处理弹窗和警告框截图和日志记录多窗口和多标签页操作一个实战的小demo步骤一&#…...

编程日记 2025/5/9 1:30:31

Ubuntu中批量重命名，rename

你可以使用下面的命令批量重命名这些文件，在文件名中插入 _1： 方式一使用 mv 命令批量重命名如果你已经在终端中，且当前目录包含这些文件，可以执行以下命令： mv ai.c ai_1.c mv ai.h ai_1.h mv ao.c ao_1.c mv a…...

编程日记 2025/5/11 20:49:09

LINUX 下 NODE 安装与配置

一、官网地址： （中文网）https://nodejs.cn/ （英文网）https://nodejs.org/en/ 二、下载安装包 2.1、下载地址：下载 | Node.js 中文网 https://nodejs.cn/download/ 2.2、使用 wget 命令下载到linux 服务器…...

编程日记 2025/4/30 20:20:07

Vue3 el-tree-v2渲染慢的问题

一、现象使用el-tree-v2处理组织架构权限时，整个树的数据在8500条，勾选数据8200条，打开页面需要8~10秒，用户无法接受。经调试，发现主要卡在树的渲染回显上（勾选数据少时，很快，勾选…...

编程日记 2025/5/11 0:27:19

【redis初阶】浅谈分布式系统

目录一、常见概念 1.1 基本概念 2.2 评价指标（Metric） 二、架构演进 2.1 单机架构 2.2 应用数据分离架构 2.3 应用服务集群架构 2.4 读写分离/主从分离架构 2.5 引入缓存 ⸺ 冷热分离架构 2.6 数据库分库分表 2.7 业务拆分 ⸺ 引入微服务 redis学习&…...

编程日记 2025/5/9 16:00:58

模式识别与机器学习 | 十一章概率图模型基础

隐马尔科夫模型（Hidden Markov Model,HMM） HMM是建模序列数据的图模型 1、第一个状态节点对应一个初始状态概率分布 2、状态转移矩阵A, 3、发射矩阵概率B 4、对特定的（x,y）的联合概率可以表示为 α递归计算——前向算法β递归…...

编程日记 2025/5/12 4:52:14

Linux基本指令（1）

一、ls指令功能：对于目录，显示这个目录下的目录名以及文件名；对于文件，显示文件名后面可接命令行选项配合使用，接选项时ls与选项以及选项与选项之间要有一个空格； 这里先学习了两个选项：-l…...

编程日记 2025/5/8 2:56:22

逐“绿”前行企业综合能源管控低碳转型如何推进？

引言： 在“双碳”战略指引下，中国低碳节能各项工作有序推进，逐步建立起碳达峰碳中和“1N”的政策体系，重点领域、重点行业及各地区的碳达峰实施方案相继出台。能源对于促进经济社会发展、增进人民福祉至关重要。近年来&#xff0…...

编程日记 2025/5/12 21:37:49

springboot和vue配置https请求

项目场景： 代码发布到线上使用https请求需要配置ssl证书，前后端都需要修改。问题描述如图，我们在调用接口时报如下错误，这就是未配置ssl但是用https请求产生的问题。解决方案： 前端：在vite.config.js文…...

编程日记 2025/5/11 15:58:52

数据库（2）--建表表操作

1.建表语法： create table if not exists 表名( 类型名类型 comment ‘注释内容’, ... )设置字符集编码与排序规则; create table if not exists student( name char(10) comment 姓名, id bigint comment 学号 )character set utf8mb4 collate utf8mb4_0900_a…...

编程日记 2025/5/5 17:36:09

泷羽sec----学会并玩转powershell【基础1-2】

声明！ 学习视频来自B站up主泷羽sec 有兴趣的师傅可以关注一下，如涉及侵权马上删除文章，笔记只是方便各位师傅的学习和探讨，文章所提到的网站以及内容，只做学习交流，其他均与本人以及泷羽sec团队无关&#…...

编程日记 2025/5/11 17:51:21

2 逻辑符号

在文件和目录的判断或者其他情况中，可以组合使用多个条件。逻辑与 (&&) 逻辑与运算符 && 用于在多个条件都为真时执行某个操作。 # 判断文件是否存在且可读 if [ -f "$file" ] && [ -r "$file" ]; thenecho "…...

编程日记 2025/5/12 15:15:30

Android基于回调的事件处理

Android 中的回调机制：基于回调的事件处理详解在 Android 开发中，回调（Callback）是一种常见的事件处理机制，主要用于异步操作和事件通知。与传统的基于监听器的事件处理相比，回调机制更加灵活、通用&…...

编程日记 2025/4/28 2:23:22

跨界融合：人工智能与区块链如何重新定义数据安全？

引言：数据安全的挑战与现状在信息化驱动的数字化时代，数据已成为企业和个人最重要的资产之一。然而，随着网络技术的逐步优化和数据量的爆发式增长，数据安全问题也愈变突出。数据安全现状：– 数据泄露驱动相关事件驱…...

编程日记 2025/5/1 1:28:49

qml SpringAnimation详解

1. 概述 SpringAnimation 是 Qt Quick 中用于模拟弹簧效果的动画类。它通过模拟物体在弹簧力作用下的反应，产生一种振荡的动画效果，常用于模拟具有自然回弹、弹性和振动的动态行为。这种动画效果在 UI 中广泛应用，特别是在拖动、拉伸、回弹等…...

编程日记 2025/5/2 2:20:39

Qt 5.14.2 学习记录 —— 칠 QWidget 常用控件（2）

文章目录 1、Window Frame2、windowTitle3、windowIcon4、qrc机制5、windowOpacity 1、Window Frame 在运行Qt程序后，除了用户做的界面，最上面还有一个框，这就是window frame框。对于界面的元素，它们的原点是Qt界面的左上角或win…...

编程日记 2025/5/7 9:55:03

Windows service运行Django项目

系统：Windows Service 软件：nssm，nginx 配置Django项目 1、把Django项目的静态文件整理到staticfiles文件夹中注：settings中的设置 STATIC_URL /static/ STATIC_ROOT os.path.join(BASE_DIR, staticfiles/) STATICFILES_DI…...

编程日记 2025/5/9 10:16:15

notebook主目录及pip镜像源修改

目录一、notebook主目录修改二、pip镜像源修改一、notebook主目录修改在使用Jupyter Notebook进行数据分析时，生成的.ipynb文件默认会保存在Jupyter的主目录中。通常情况下，系统会将Jupyter的主目录设置为系统的文档目录，而文档目录通常位…...

编程日记 2025/5/7 17:09:10

Linux(Centos 7.6)命令详解：touch

1.命令作用如果文件不存在将创建一个空文件；修改文件修改访问时间。 2.命令语法 Usage: touch [OPTION]... FILE... 3.参数详解 Usage: -a，access 只修改访问时间-c，不创建任何文件-d，--dateSTRING 解析STRING并使用它替代…...

编程日记 2025/5/13 1:14:21

ubuntu 下生成 core dump

在Ubuntu下，发现程序崩溃后不生成core dump文件, 即使设置了ulimit -c unlimited后仍然无效。 1.ulimit -c unlimited 输出的的含义是核心转储文件的大小限制，单位是blocks，默认是0，表示不生成core dump文件。 2. 重设core_pattern ulimit -c unlimited后，核心转储文件…...

编程日记 2025/5/11 13:18:49

大数据智能选课系统

1.产品介绍产品名称：大数据智能选课系统一、产品概述随着信息技术的快速发展，大数据技术在教育领域的应用越来越广泛。针对当前高校选课过程中的繁琐操作、资源分配不均等问题，我们研发了一款基于大数据智能分析的选课系统。本系统旨在…...

编程日记 2025/5/12 0:08:51

HTTP-响应协议

HTTP的响应过程？ 浏览器请求数据--》web服务器过程：请求过程 web服务器将响应数据-》到浏览器：响应过程响应数据有哪些内容？ 1.和请求数据类似。 2. 响应体中存储着web服务器返回给浏览器的响应数据。并且注意响应头和响应体之间…...

编程日记 2025/5/13 1:47:13

[离线数仓] 总结三、Hive数仓DIM层开发

5.9 数仓开发之DIM层 DIM层设计要点：（1）DIM层的设计依据是维度建模理论，该层存储维度模型的维度表。（2）DIM层的数据存储格式为orc列式存储+snappy压缩。（3）DIM层表名的命名规范为dim_表名_全量表或者拉链表标识（full/zip）。 -- 数仓开发之DIM层 -- DIM层设计要点：…...

编程日记 2025/5/8 7:06:21

搭建RK3588开发板Qt交叉编译环境

一、开发环境在虚拟机里安装Ubuntu20.04Qt5.14.2交叉编译器gcc-linaro-7.5.0 二、相关资料下载在虚拟机里安装Ubuntu20.04 Ubuntu20.04镜像下载（https://mirrors.tuna.tsinghua.edu.cn/ubuntu-releases/）安装Qt5.14.2 下载安装包和源码安装包(http…...

编程日记 2025/5/11 17:52:32

【Rust自学】11.3. 自定义错误信息

喜欢的话别忘了点赞、收藏加关注哦，对接下来的教程有兴趣的可以关注专栏。谢谢喵！(･ω･) 11.3.1. 添加错误信息在 11.2. 断言(Assert) 中我们学习了assert!、assert_eq!和assert_ne!这三个宏，而这篇文章讲的就是它…...

编程日记 2025/5/7 20:07:17

ROS核心概念解析：从Node到Master，再到roslaunch的全面指南

Node 在ROS中，最小的进程单元就是节点（node）。一个软件包里可以有多个可执行文件，可执行文件在运行之后就成了一个进程(process)，这个进程在ROS中就叫做节点。从程序角度来说，node就是一个可执行文件&…...

编程日记 2025/5/10 0:53:32

Autodl安装tensorflow2.10.0记录

首先租用新实例（我选的是3080*2卡），由于基础镜像中没有2.10.0版本，选miniconda3的基础环境创建虚拟环境：conda create --name xxx python3.8（环境名）激活虚拟环境：conda activate x…...

编程日记 2025/5/6 10:26:52

Linux第一课：c语言学习记录day06

四、数组冒泡排序两两比较，第 j 个和 j1 个比较 int a[5] {5, 4, 3, 2, 1}; 第一轮：i 0 n：n个数，比较 n-1-i 次 4 5 3 2 1 // 第一次比较 j 0 4 3 5 2 1 // 第二次比较 j 1 4 3 2 5 1 // 第三次比较 j 2 4 3 2 1 5 // …...

编程日记 2025/5/8 17:22:50

代码随想录day03

203 链表基础操作 class Solution { public:ListNode* removeElements(ListNode* head, int val) {while (head!NULL&&head->valval){ListNode* temphead;headhead->next;delete temp;}ListNode* curhead;while (cur!NULL&&cur->next!NULL){if(cur-…...

编程日记 2025/5/13 1:31:53