I have the same issue, being that my data is of size torch.Size([32, 3, 64, 64]), and I’m using the transformations:
transform = transforms.Compose(
    [transforms.RandomHorizontalFlip(p=0.2),
    transforms.ColorJitter(brightness=.5, hue=.1),
    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])

I am calling as this:
transform(data)

According to the docs, this composition should accept tensor. My PyTorch version is 1.7.1
Why is there even such difference between accepting PIL or tensor types? This makes everything so much harder