pytorch_scatter could maybe be faster for features aggregation

[tchaton]

https://github.com/rusty1s/pytorch_scatter

[mys007]

Sorry for the massive delay and thanks for making me aware of this framework. It's definitely worth a try, as well as looking into https://github.com/dmlc/dgl/ for a re-implementation....

[tchaton]

I think pytorch_geometric should also be an option.

[mys007]

Indeed, in fact, it seems that they've even implemented ECC themselves: https://github.com/rusty1s/pytorch_geometric/blob/master/torch_geometric/nn/conv/nn_conv.py !