Display number of images per class using Pytorch

Question

I am using Pytorch with FashionMNIST dataset I would like to display 8 image sample from each of the 10 classes, however, I did not figure how to split the training test int