Convert 3d vector of axis-angle rotation to 3x3 rotation matrix. Shape: - Input: :Torch:`(N, 3)` - Output: :Torch:`(N, 3, 3)`
(angle_axis)
| 57 | return axis_angle |
| 58 | |
| 59 | def axis_angle_to_mat3x3(angle_axis): |
| 60 | """ |
| 61 | Convert 3d vector of axis-angle rotation to 3x3 rotation matrix. |
| 62 | Shape: |
| 63 | - Input: :Torch:`(N, 3)` |
| 64 | - Output: :Torch:`(N, 3, 3)` |
| 65 | """ |
| 66 | rot_mat = tgm.angle_axis_to_rotation_matrix(angle_axis) # 4x4 rotation matrix |
| 67 | |
| 68 | return rot_mat[:, :3, :3] |
| 69 | |
| 70 | def mat3x3_to_axis_angle(rot_mat): |
| 71 | """ |
no outgoing calls
no test coverage detected