- 基类 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)