deepfold.losses.auxillary.chain_centre_mass_loss

deepfold.losses.auxillary.chain_centre_mass_loss(pred_atom_positions: Tensor, true_atom_positions: Tensor, atom_mask: Tensor, asym_id: Tensor, eps: float = 1e-10) Tensor[source]