deepfold.losses.geometry.superimpose

deepfold.losses.geometry.superimpose(src_atoms: Tensor, tgt_atoms: Tensor, mask: Tensor) Tuple[Tensor, Tensor][source]

Superimposes coordinates onto a tgt_atoms by minimizing RMSD using SVD.

Parameters:
  • src_atoms – reference tensor shaped [*, N, 3]

  • tgt_atoms – target tensor shaped [*, N, 3]

  • mask – mask tensor shaped [*, N]

Returns:

superimposed coords [, N, 3] rmsds: final RMSDs []

Return type:

superimposed