import torch
import torchvision
from torchvision import transforms, datasets
train = datasets.MNIST("", train=True, download=True,
transform=transforms.Compose([transforms.ToTensor()]))
test = datasets.MNIST("", train=False, download=True,
transform=transforms.Compose([transforms.ToTensor()]))