I have a rotation matrix:
rotation_A = torch.tensor([[[ 0.7198, -0.6428, -0.2620],
[ 0.5266,  0.7516, -0.3973],
[ 0.4523,  0.1480,  0.8795]]])
and I want to rotate a cube with a size of torch.Size([74, 626, 766]).
How can I do that with PyTorch? or simple python?