torch 默认参数初始化_torch_geometric 源码阅读（一）

最近摸鱼严重，尝试写博客逼迫自己学习（x （主要给自己看，可能不太说人话hhh，部分细节待补充。）

torch_geometric.nn-Convolutional Layers

MessagePassing

源码部分刨去对GNNExplainer的支持，核心的几行如下，其中关键的函数为 message, aggregate, update。message & update 可简单理解成分别对图卷积网络的输入输出做操作，例如添加线性层降维；而 aggregate 为将某节点周围邻居节点的 feature 进行聚合。

coll_dict = self.__collect__(self.__user_args__, edge_index, size, kwargs)
            
msg_kwargs = self.inspector.distribute('message', coll_dict)    
out = self.message(**msg_kwargs)

aggr_kwargs = self.inspector.distribute('aggregate', coll_dict)
out = self.aggregate(out, **aggr_kwargs)

update_kwargs = self.inspector.distribute('up