I am doing some project on object reconstruction and I need to rotate object in PyTorch while training to make sure I am augmenting data properly. My data consists of points and it's normals. My question is, how should I (or should I traing them at all) rotate normals of my mesh. Note that this rotation also needs to keep object in unit cube, where it was before running this code.

I am doing some project on object reconstruction and I need to rotate an object in PyTorch during the training to make sure I am augmenting the data properly. My data consists of points and it's normals. My question is, how should I (or should I train them at all) rotate the normals of my mesh. Note that this rotation also needs to keep the object inside the unit cube, where it was before the transformation.