While there is an example of the DiffPool paper, it is not clear how to use with existing nn layers like GraphConv or RelGraphConv. The example uses a custom SageGraph model which explicitly takes the features and adgacency matrix as inputs and goes on to further so the message passing.
The diffPooling operation generates an updated feature matrix and an updated adgacency matrix.
With the current nn modules is there a way of sending the updated adgacency matrix for a batched graph input such that the message update and apply funcrion work on basis of this new structural information.