class_docstr: Helper class to contain schema for loss hybrid_forward input
class_docstr_tok: ["Helper","class","to","contain","schema","for","loss","hybrid_forward","input"]
docstr: :param head_outputs: list of argument names in hybrid_forward that are outputs of the head.
      The order and number MUST MATCH the output from the head.
  :param agent_inputs: list of argument names in hybrid_forward that are inputs from the agent.
      The order and number MUST MATCH `output_<head_type_idx>_<order>` for this head.
  :param targets: list of argument names in hybrid_forward that are targets for the loss.
      The order and number MUST MATCH targets passed from the agent.
docstr_tok: [":","param","head_outputs",":","list","of","argument","names","in","hybrid_forward","that","are","outputs","of","the","head",".","The","order","and","number","MUST","MATCH","the","output","from","the","head",".",":","param","agent_inputs",":","list","of","argument","names","in","hybrid_forward","that","are","inputs","from","the","agent",".","The","order","and","number","MUST","MATCH","`","output_","<","head_type_idx",">","_","<","order",">","`","for","this","head",".",":","param","targets",":","list","of","argument","names","in","hybrid_forward","that","are","targets","for","the","loss",".","The","order","and","number","MUST","MATCH","targets","passed","from","the","agent","."]
code: def __init__(
      self,
      head_outputs: List[str],
      agent_inputs: List[str],
      targets: List[str],
  ):
      """
      :param head_outputs: list of argument names in hybrid_forward that are outputs of the head.
          The order and number MUST MATCH the output from the head.
      :param agent_inputs: list of argument names in hybrid_forward that are inputs from the agent.
          The order and number MUST MATCH `output_<head_type_idx>_<order>` for this head.
      :param targets: list of argument names in hybrid_forward that are targets for the loss.
          The order and number MUST MATCH targets passed from the agent.
      """
      self._head_outputs = head_outputs
      self._agent_inputs = agent_inputs
      self._targets = targets
  
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 LossInputSchema object from inherited base classes: object with example: obj = LossInputSchema(head_outputs, agent_inputs, targets)