I want to import diffpool using DGL, but I am unsure whether the gradient will flow through the creation of dgl.Graph()
objects. For instance, in the paper at equation 4 we make a new adjacency for the coarsened graph. This adjacency calculation comes through some tensors that should have gradients attached to them; if I then used the adjacency to make a new dgl.Graph()
object to then use in a GNN further downstream, would the gradients from the adjacency be preserved or not?
It might be that I just have to implement everything manually.