deepfold.utils.rigid_utils.rot_matmul

deepfold.utils.rigid_utils.rot_matmul(a: Tensor, b: Tensor) → Tensor

Performs matrix multiplication of two rotation matrix tensors. Written out by hand to avoid AMP downcasting.

Parameters:

• a – [*, 3, 3] left multiplicand

• b – [*, 3, 3] right multiplicand

Returns:

The product ab