class_docstr: Dice loss for segmentation
docstr: Activation function applied to model output
docstr_tok: ["Activation","function","applied","to","model","output"]
code: def activation(self, x: Tensor) -> Tensor:
      "Activation function applied to model output"
      return F.softmax(x, dim=self.axis)
  
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 4 for calling the DiceLoss obj's underlying member method code with example usage: obj.activation(x) and returns: F