You can use affine transformations like rotation as data augmentation.
I’m still not clear, if I misunderstand the question, but since you passed the transformations to your Dataset, they will be used for each sample (also after splitting the Dataset).