@Fnjn Umm, I am not working on data augmentation. It’s something on a custom function.
EDIT: But I think I could use the source code of torchvision.transforms.RandomRotation for reference. Thanks a lot.