Skip to main content

Module gnn

Module gnn 

Source
Expand description

GNN(图神经网络)原语模块

提供图神经网络的核心构建块:

  • 消息传递框架
  • 图卷积层(GCN, GAT, GraphSAGE)
  • 图 pooling 和 normalization

§示例

use god_gragh::tensor::gnn::{MessagePassingLayer, GCNConv, SumAggregator};
use god_gragh::tensor::DenseTensor;

// 创建 GCN 层
let gcn = GCNConv::new(64, 64);

// 前向传播
let output = gcn.forward(&node_features, &adjacency);

Structs§

GATConv
GAT(图注意力网络)层
GCNConv
GCN(图卷积网络)层
GraphSAGE
GraphSAGE 层
IdentityMessage
恒等消息函数:直接传递源节点特征
LinearMessage
线性消息函数:应用线性变换
MaxAggregator
最大值聚合器
MeanAggregator
均值聚合器
MessagePassingLayer
消息传递层:GNN 的核心构建块
SumAggregator
求和聚合器

Traits§

Aggregator
聚合器 trait:定义邻居消息的聚合方式
MessageFunction
消息函数 trait:定义边上的消息计算
UpdateFunction
更新函数 trait:定义节点状态更新