I want to Normalize the MNIST dataset with pyTorch. The code I was given to load the dataset is:
mnist_train = datasets.MNIST(data_dir, download=True, train=T