I am trying to divide a graph dataset into multiple batches using DataLoader. My input data is consist of a graph signal of size `[B, feature * node]`

and graph adjacency matrix of size `[B, node, node]`

. I was wondering if there is a way to divide both tensor into batches at once using DataLoader.

Found the solution here:

1 Like