change formatting

Author: pri1311

torchvision/ops/dice_loss.py

@@ -1,8 +1,9 @@
 import torch
 import torch.nn.functional as F
 
+
 def dice_loss(inputs: torch.Tensor, targets: torch.Tensor, reduction: str = "none", eps: float = 1e-8) -> torch.Tensor:
-    """Criterion that computes Sørensen-Dice Coefficient loss.
+    r"""Criterion that computes Sørensen-Dice Coefficient loss.
 
     We compute the Sørensen-Dice Coefficient as follows:
 
@@ -71,4 +72,4 @@ def dice_loss(inputs: torch.Tensor, targets: torch.Tensor, reduction: str = "non
             f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'"
         )
 
-    return loss
+    return loss