import torch import torchvision import torchvision.transforms as transforms