I use the MNIST dataset to learn Pytorch.
This is from the documentation to get a picture.
# Create prediction images, labels = next(iter(trainloader)) ima