tomodrgn.lie_tools.so3_entropy

so3_entropy(w_eps: Tensor, std: Tensor, k: int = 10) → Tensor

Parameters:

• w_eps – (Tensor of dim Bx3): sample from so3

• std – (Tensor of dim Bx3): std of distribution on so3

• k – Use 2k+1 samples for truncated summation

Returns: