Fix inconsistant cuda type in utils and so3

[poptree]

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.