tomodrgn.lie_tools.random_SO3

random_SO3(n: int, dtype: dtype = torch.float32, device: device | None = None) → Tensor

Generate random rotations as SO3 matrices

Parameters:

• n – number of rotations to generate

• dtype – dtype of output tensor

• device – device for output tensor

Returns:

random SO3 rotations tensor