tomodrgn.lie_tools.random_quaternions#

random_quaternions(n: int, dtype: dtype = torch.float32, device: device | None = None) Tensor[source]#

Generate random rotations as quaternions

Parameters:
  • n – number of rotations to generate

  • dtype – dtype of output tensor

  • device – device for output tensor

Returns:

random quaternions tensor