deepfold.utils.tensor_utils.masked_mean

deepfold.utils.tensor_utils.masked_mean(mask: Tensor, value: Tensor, dim: int | Tuple[int, ...], eps: float = 0.0001, keepdim: bool = False) → Tensor