deepfold.utils.rigid_utils.rot_vec_mul

deepfold.utils.rigid_utils.rot_vec_mul(r: Tensor, t: Tensor) → Tensor[source]

Applies a rotation to a vector. Written out by hand to avoid transfer to avoid AMP downcasting.

Parameters:

  • r – [*, 3, 3] rotation matrices

  • t – [*, 3] coordinate tensors

Returns:

[*, 3] rotated coordinates