dgl.ops.gspmm
- dgl.ops.gspmm(g, op, reduce_op, lhs_data, rhs_data)[source]
广义稀疏矩阵乘法接口。 它将两个步骤融合到一个内核中。
通过
op源节点和边特征计算消息。通过
reduce_op聚合消息作为目标节点的特征。
\[x_v = \psi_{(u, v, e)\in \mathcal{G}}(\rho(x_u, x_e))\]其中 \(x_v\) 是目标节点上返回的特征，而 \(x_u\), \(x_e\) 分别指代
u,e。 \(\rho\) 表示二元运算符op，而 \(\psi\) 表示归约运算符reduce_op, \(\mathcal{G}\) 是我们应用 gspmm 的图:g。请注意，此函数不处理梯度。