class_docstr: ABC for loss functions of each head. Child class must implement input_schema() and loss_forward()
class_docstr_tok: ["ABC","for","loss","functions","of","each","head",".","Child","class","must","implement","input_schema","(",")","and","loss_forward","(",")"]
code: def __init__(self, *args, **kwargs):
      super(HeadLoss, self).__init__(*args, **kwargs)
      self._output_schema = None
  
imports: from typing import Dict, List, Union, Tuple
  import mxnet
  from mxnet.initializer import Initializer, register
  from mxnet.gluon import nn, loss
  from mxnet.ndarray import NDArray
  from mxnet.symbol import Symbol
  from rl_coach.base_parameters import AgentParameters
  from rl_coach.spaces import SpacesDefinition
label_desc: Use image node_id 1 to create a new HeadLoss object from inherited base classes: loss with example: obj = HeadLoss()