tomodrgn.lie_tools.SO3_to_quaternions

SO3_to_quaternions(r: Tensor) → Tensor

Map batch of SO(3) matrices to quaternions.

Parameters:

r – SO3 rotation matrices

Returns:

equivalent quaternions