I think your new transformation won’t work, as you are trying to apply image transformations (e.g. RandomRotation) on tensors.
Instead you could apply FiveCrop, loop over each cropped image, and apply the rotation to it.
I’m not sure what size you are referring to in the last sentence. CIFAR-10 images have a spatial size of 32x32. You could of course pad them to 40x40 and check, if that helps.