I want implement augmentation in a 3D dataset with an tensorflow model.
the augmentation function is like this: