trans =  transforms.Compose([
transforms.ToPILImage(),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
i want to apply this transform to 4d tensor (n, 3, 224, 224) and get a tensor in return.