I would like to calculate pairwise dot products across every node in each graph in a batch. I was going to make the graph complete and use apply_edges, but completing the graph by looping through it seemed cumbersome. My other thought was just keeping track of all the sub graphs in the batch and dot together their node features.
Does anyone have any slicker ways to do it?
Thank you!