本文详细介绍了PyTorch_Geometric库中的MessagePassing基类，它是构建各种图卷积网络的基础。MessagePassing涉及到节点特征的聚合、信息传递和更新，通过message()和update()函数实现。文章还给出了GCNs的具体实现步骤，并讨论了数据结构在邻接矩阵乘以属性矩阵时的处理方式。
- 基类 message passing layers，其他卷积都是以这个类为基类构建的。
- 描述：分层聚合每一个节点与周围节点及关联边信息的通用函数。
- 所在位置：torch_geometric.nn.MessagePassing.
- 初始化：
__init__(self, aggr='add', flow='source_to_target')
这个基类的功能可以形式化为下面的公式：
x i ′ = γ Θ ( x i , □ j ∈ N ( i )   ϕ Θ ( x i , x j , e i , j ) ) , \mathbf{x}_i^{\prime} = \gamma_{\mathbf{\Theta}} \left( \mathbf{x}_i, \square_{j \in \mathcal{N}(i)} \, \phi_{\mathbf{\Theta}} \left(\mathbf{x}_i, \mathbf{x}_j,\mathbf{e}_{i,j}\right) \right), xi′=γΘ(xi,□j∈N(i)ϕΘ(xi,xj,ei,j)),
-
x i ( k − 1 ) ∈ R F \mathbf{x}^{(k-1)}_i \in \mathbb{R}^F xi(k−1)
最低0.47元/天 解锁文章
273
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
