Python Torchvision.transforms.compose