1 前言
此文为自己学习的总结,内容多有参考其他文章内容,并在参考文献处一并给出文章链接。
2 介绍
Pytorch Geometric和dgl是两个基于Message Passing(消息传递)的图神经网络框架,两个框架的实现方式个人感觉有很大的不同。在此处介绍pyg的消息传递方式。
图中的卷积计算通常被称为邻域聚合或者消息传递 (neighborhood aggregation or message passing). 定义 x i ( k − 1 ) ∈ R F \mathbf{x}_i ^{( k − 1 )} \in R^F xi(k−1)∈RF为节点 i i i在第 ( k − 1 ) (k-1) (k−1) 层的特征

本文介绍了PyTorchGeometric和DGL这两个基于消息传递的图神经网络框架。重点阐述了PyTorchGeometric中的MessagePassing类,解释了消息传递的三步过程,包括消息生成、聚合和节点更新,并通过实例展示了如何自定义卷积操作。此外,还提供了相关参考资料以深入理解GNN的实现。
最低0.47元/天 解锁文章
5813

被折叠的 条评论
为什么被折叠?



