tomodrgn.lie_tools.SO3_to_s2s2#

SO3_to_s2s2(r: Tensor) → Tensor[source]#

Map batch of SO(3) matrices to s2s2 representation as first two basis vectors, concatenated as Bx6