You cannot rotate an image inside of save_image() [1] 1, but you can use the method
torchvision.transforms.functional.roate(img, angle, resample=False, expand=False, center=None)

to apply transformations to PIL images, while using PyTorch [2] 2.