Using DataLoader to load two tensors

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