Dave Schnizlein cd5af2521a Update Rotate transform to use device of input rotation ...

Summary: Currently the Rotate transform does not consider the R's device at all, resulting in errors if you're expecting it to be on cuda but it gets the default casting to cpu. This updates the transform to respect R's device.

Reviewed By: nikhilaravi

Differential Revision: D27828118

fbshipit-source-id: ddd99f73eadbd990688eb22f3d1ffbacbe168c81

2021-05-19 10:04:12 -07:00

..

__init__.py

Expose axis-angle/quaternion conversion functions in transforms package (#486)

2020-12-18 05:18:48 -08:00

rotation_conversions.py

spelling

2021-04-09 09:58:54 -07:00

so3.py

6D representation of rotations.

2020-07-08 04:01:22 -07:00

transform3d.py

Update Rotate transform to use device of input rotation

2021-05-19 10:04:12 -07:00