How to calculate loss and gradient of index in Pytorch

Question

I have a misaligned point cloud and rotation matrix and I want to use backprop to tune the rotation matrix in 2d plane.

The problem is in the forward all I c