GSpMM

Generalized Sparse-Matrix Dense-Matrix Multiplication operators, which are proved to be the building blocks in GNN. Here we only implement the forward propagation of GSpMM. It functions is as follows. 1. Computes messages by add/sub/mul/div source node and edge features, or copy node features to edges.

  1. Aggregate the messages by sum/max/min/mean as the features on destination nodes.

The following are all supported operations: