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 层
- Identity
Message - 恒等消息函数:直接传递源节点特征
- Linear
Message - 线性消息函数:应用线性变换
- MaxAggregator
- 最大值聚合器
- Mean
Aggregator - 均值聚合器
- Message
Passing Layer - 消息传递层:GNN 的核心构建块
- SumAggregator
- 求和聚合器
Traits§
- Aggregator
- 聚合器 trait:定义邻居消息的聚合方式
- Message
Function - 消息函数 trait:定义边上的消息计算
- Update
Function - 更新函数 trait:定义节点状态更新