I have a BatchedDGLGraph, each graph with a set of nodes which, after some operations, I classify as 0 / 1. Is there a way to group all ‘0’, ‘1’ nodes and then split them into their own graphs? So that after each iteration, the batch size doubles (though the number of nodes remains constant).

Thanks!