Hey @Bhavya_Soni
Why would you concatenate the augmented tensors?
It would be efficient to augment and send the augmented tensors through a model and learn the weights.
The only problem that I see here is that the Random_ augmentation will augment each image in a batch the same way, which is not an ideal setup.