find more:-  3

import torchvision
import torchvision

import MNIST datasets and transforms
from torchvision.datasets import MNIST
from torchvision import transforms
transform = transforms.Compose([transforms.ToTensor(),
transforms.Normalize((0.5,), (0.5,)),
])

download training dataset
train_ds = MNIST(root=“data”, download=True, transform=transform)