Hey @Ajinkya_Ambatwar
I have made a GIST 6 that takes an image and transforms is with RandomAffine(10). This indeed randomly transforms each image in the batch with a random degree.
I would also like to add that, in the code snippet you have provided, I guess you have mistyped utils.DataLoader, it should have been utils.data.DataLoader. Another little thing to add here is that a DataSet has a transform parameter. The GIST provided has all that you would need for an ablation study.
Hope this might help you 

Screen-Shot

image532×1122 97.3 KB