Fixing device selection (cuda or cpu)

[ArashJavan]

The class methods in so3 creates new matrices (eye etc.) without taking the device selection taking into account. This patch should fix this by setting the right device directly while creating new matrices.