tomodrgn.lie_tools.so3_entropy_old#

so3_entropy_old(w_eps: Tensor, std: Tensor, k: int = 10) Tensor[source]#
Parameters:
  • w_eps – (Tensor of dim 3): sample from so3

  • std – (Tensor of dim 3x3): covariance of distribution on so3

  • k – 2k+1 samples for truncated summation

Returns: