(quaternions)
| 312 | |
| 313 | |
| 314 | def quaternion_to_cont6d(quaternions): |
| 315 | rotation_mat = quaternion_to_matrix(quaternions) |
| 316 | cont_6d = torch.cat([rotation_mat[..., 0], rotation_mat[..., 1]], dim=-1) |
| 317 | return cont_6d |
| 318 | |
| 319 | |
| 320 | def cont6d_to_matrix(cont6d): |
no test coverage detected