How to sample along with another dataloader in Pytorch

Question

Assume I have train/valid/test dataset with batch_size and shuffleed as normal.

When I do train/valid/test, I want to sample a certain number (c