Torchvision.transforms.compose