3D augmentation function in Tensorflow 2.0

Question

I haven\'t found any easy to use augmentation in Tensorflow 2.0x for 3D data. Currently, I am using a library called volumentations, and my code is below using