deepfold.losses.confidence.tm_loss

deepfold.losses.confidence.tm_loss(logits: Tensor, final_affine_tensor: Tensor, backbone_rigid_tensor: Tensor, backbone_rigid_mask: Tensor, resolution: Tensor, max_bin: int = 31, num_bins: int = 64, min_resolution: float = 0.1, max_resolution: float = 3.0, eps: float = 1e-08) → Tensor