news 2026/4/16 11:40:39

图神经网络与pytorch

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
图神经网络与pytorch

目录

一、图神经网络架构pytorch

1.GCNConv

(1)、数学公式(标准 GCN)

(2)一个例子

(3)、解释一下

2.归一化层

(1).定义

(2).一个例子

3.GATConv

(1)、如何计算的

🕐计算原始注意力系数:

(2)、多头注意力机制


一、图神经网络架构pytorch

1.GCNConv

self.conv1 = GCNConv(in_dim, hidden_dim)

(1)、数学公式(标准 GCN)

(2)一个例子

图结构内容如下

邻接矩阵 A(不含自环)

计算归一化系数

假设每个节点 1 维特征(为了好算):

线性变换权重 W

先算 XW

之后有

(3)、解释一下

对于每个结点来说,他的来源包括自己以及和自己相连的其他结点的内容。归一化就是让自己的数据/根号下结点的度 + 其他数据/根号下结点的度。但是由于度越大,结合的数据越多,所以还要在外面进行归一化,/根号下结点的度。

第一步计算每个结点和权重矩阵处理后的结果

第二步是每个结点的数据与他相连的数据,归一化后求和

第三步是 对计算的结果进行最后的归一化

2.归一化层

self.norm1 = nn.LayerNorm(hidden_dim)

不跨节点、不跨 batch、不关心图大小

(1).定义

(2).一个例子

假设某个节点经过 GCNConv 后:

h_i = [100, 0.1, -3, 2, 50]

这会导致:

  • 后面的 ReLU / Linear 非常不稳定

  • PPO 的 value / logit 很容易炸

LayerNorm 会把它拉成:

h_i ≈ [1.2, -0.8, -1.1, -0.6, 1.3]

👉形状不变,尺度统一

3.GATConv

self.conv1 = GATConv(in_dim, hidden_dim, heads=heads, concat=True) # 注意:concat=True 表示将每个头的输出拼接

(1)、如何计算的

🕐计算原始注意力系数

🕑对所有邻居进行计算

对于节点i的每个邻居节点j,都需要计算上述的注意力系数。通过这种方式,我们可以得到所有邻居节点的注意力系数。注意这个邻居也包括他自己

🕑注意力系数的归一化

🕑加权聚合邻居特征

(2)、多头注意力机制

把上面的过程复制几次,那么每个结点都可以得到很多个维度一样的特征,把这个特征拼接在一起,就是多头注意力机制。

下图中一条颜色的线,就可以表示一个注意力头。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/16 15:14:06

鸿蒙HarmonyOS多线程编程实战:AI语音

在智能终端设备日益普及的今天,AI 语音应用——如语音助手、实时转写、声纹识别等——已成为用户交互的重要入口。然而,这类应用对响应速度、流畅性和系统资源调度提出了极高要求。若处理不当,极易出现卡顿、延迟甚至崩溃,严重影响…

作者头像 李华
网站建设 2026/4/15 16:35:08

机器人操作空间速度计算python几种实现函数

def compute_V0_screw_motion(x, R, xd, Rd, max_lin_vel0.05, max_ang_vel0.1, beta1.0):"""基于螺旋运动理论计算引导速度,更符合物理运动参数:beta: 收敛速度系数"""# 计算位置误差pos_error xd - x# 计算姿态误差&am…

作者头像 李华
网站建设 2026/4/15 15:27:36

基于SpringBoot的企业合同管理系统设计与实现(程序+文档+讲解)

课题介绍在企业合同规范化管控、全生命周期管理需求升级的背景下,传统合同管理存在 “版本混乱、审批流程长、履约监控缺失” 的痛点,基于 SpringBoot 构建的企业合同管理系统,适配合同专员、法务、部门负责人、管理层等多角色,实…

作者头像 李华
网站建设 2026/4/16 12:59:05

28、Linux 文件共享与备份全攻略

Linux 文件共享与备份全攻略 1. NFS 文件共享 NFS(网络文件系统)是一种在网络中共享文件的有效方式,使用 NFS 共享文件主要涉及两个基本步骤: - 在 NFS 服务器上,将一个或多个目录列在 /etc/exports 文件中,并运行 /usr/sbin/exportfs 命令来导出这些目录,同时要…

作者头像 李华
网站建设 2026/4/16 12:25:34

34、Linux系统安全防护全攻略

Linux系统安全防护全攻略 1. 文件加密 若仅需对文件进行加密,且无需他人解密,可使用GPG进行对称加密。操作步骤如下: 1. 执行命令 gpg -o secret.gpg -c somefile ,GPG会提示输入密码并要求再次输入以确认。之后,GPG会使用从密码生成的密钥对文件进行加密。 2. 若要…

作者头像 李华
网站建设 2026/4/15 10:05:04

36、网络安全测试工具与互联网服务解析

网络安全测试工具与互联网服务解析 在网络安全和互联网服务的领域中,有许多强大的工具和概念值得我们去探索。下面将详细介绍一些常见的安全测试工具以及互联网服务的相关知识。 安全测试工具 在进行网络安全测试时,有很多自动化工具可供选择。这些工具的功能各有不同,有…

作者头像 李华