Im trying to work on graph model in material science where the gathering step in which adjacent nodes connected by edges are concatenated in a geometric manner. Unlike the usual gathering operation where sum and max are used, here we are using a concatenation operation where ( the order of concatenation is done in a geometric manner and it’s already a precalculated).

I have figured out the way to solve it using Pytorch alone, but I was wondering if there is any model in DGL where operation variant gathering is done.

Ideas are taken from the following paper given here (Model details on Page 4)