利用MessagePassing实现GraphSAGE(了解pyG的底层逻辑)
一些基础的理解参考(必看前面的必要了解)
pyG利用MessagePassing实现GCN(了解pyG的底层逻辑)_山、、、的博客-CSDN博客pyG利用MessagePassing实现GCNhttps://blog.csdn.net/qq_44689178/article/details/123736686
注: 如果看明白上面GCN的实现,下面这个可以不看,很简单。
下面解析如何按照GraphSAGE公式,利用MessagePassing实现SAGEConv层。
下面是官方文档:
torch_geometric.nn — pytorch_geometric 2.0.5 documentationhttps://pytorch-geometric.readthedocs.io/en/latest/modules/nn.html#torch_geometric.nn.conv.message_passing.MessagePassing.propagate
首先,参考下面链接里面graphSAGE的公式
图神经网络(7
共有 0 条评论