Heng Li
Results
1
issues of
Heng Li
The class methods in so3 use `torch.eye`. However, the `mat` might be in `cuda`. Thus, the `torch.eye` should move to `cuda` if `mat` is on `cuda`.