How can I use torch Dataloader to get Picture having same class?

Question

in my dataset, there are 6 classes and 23 pictures per class
I used torchvision.dataset to make ImageFolder and it work well.

data