Take a look at this post to check how the BatchSampler could work (it’ll pass a batch of indices to the __getitem__ method). To do so, you could create a custom sampling strategy in a derived sampler class and create the batch of indices here.
Let me know, if you get stuck and I can try to follow up with an example code snippet.