Hey, I am a fresh starter with pytorch. Strangley I cannot find anything related to this, although it seems rather simple.
I want to structure my batch  with specific examples, like all examples per batch having the same label or just fill the batch with examples of just 2 classes.
How would I do that? For me it seems the right place within the dataloader and not in the dataset? As the dataloader is responsible for the batches and not the dataset?
Is there simple minimal example?