Converting augmentation of Keras into Pytorch

[Soumya_Kundu (Soumya Snigdha Kundu)]

You can make custom transforms: Writing Custom Datasets, DataLoaders and Transforms — PyTorch Tutorials 2.4.0+cu121 documentation

For width and height shift range: How shift an image vertically or horizontally?

For zoom: Pytorch randomresizedcrop vs zoom_range in datagenerator

Reflect: torch.nn.functional.pad — PyTorch 2.4 documentation (I guess)