Customized Bach Loader

I like to micromanage each batch where it gets n1 features from folder 1, n2 features from folder 2, …
How can this be accomplished this in PyTorch?