class_docstr: Dice loss for segmentation
code: def __init__(
      self,
      axis: int = 1,  # Class axis
      smooth: float = 1e-6,  # Helps with numerical stabilities in the IoU division
      reduction: str = "sum",  # PyTorch reduction to apply to the output
      square_in_union: bool = False,  # Squares predictions to increase slope of gradients
  ):
      store_attr()
  
imports: from __future__ import annotations
  from .imports import *
  from .torch_imports import *
  from .torch_core import *
  from .layers import *
label_desc: Use image node_id 1 to create a new DiceLoss object  with example: obj = DiceLoss(axis, smooth, reduction, square_in_union)