Thanks for the update. If I understand your use case correctly you want to use a specific sampling strategy where each batch should contain a sample of each Re-index or at least they should not be repeated in a single batch.
In that case you could create a custom sampler and implement the sampling logic there to make sure each batch contains different Re-index values.