class_docstr: Base class for optimization objectives.
  
  Args:
      name (string): <span style="color:#0000C0"><b>internal use</b></span>.
      states_spec (specification): <span style="color:#0000C0"><b>internal use</b></span>.
      internals_spec (specification): <span style="color:#0000C0"><b>internal use</b></span>.
      auxiliaries_spec (specification): <span style="color:#0000C0"><b>internal use</b></span>.
      actions_spec (specification): <span style="color:#0000C0"><b>internal use</b></span>.
      reward_spec (specification): <span style="color:#0000C0"><b>internal use</b></span>.
class_docstr_tok: ["Base","class","for","optimization","objectives",".","Args",":","name","(","string",")",":","<","span","style=","''","color",":","#","0000C0","''",">","<","b",">","internal","use","<","\/b",">","<","\/span",">",".","states_spec","(","specification",")",":","<","span","style=","''","color",":","#","0000C0","''",">","<","b",">","internal","use","<","\/b",">","<","\/span",">",".","internals_spec","(","specification",")",":","<","span","style=","''","color",":","#","0000C0","''",">","<","b",">","internal","use","<","\/b",">","<","\/span",">",".","auxiliaries_spec","(","specification",")",":","<","span","style=","''","color",":","#","0000C0","''",">","<","b",">","internal","use","<","\/b",">","<","\/span",">",".","actions_spec","(","specification",")",":","<","span","style=","''","color",":","#","0000C0","''",">","<","b",">","internal","use","<","\/b",">","<","\/span",">",".","reward_spec","(","specification",")",":","<","span","style=","''","color",":","#","0000C0","''",">","<","b",">","internal","use","<","\/b",">","<","\/span",">","."]
code: def input_signature(self, *, function):
      if function == "loss":
          return SignatureDict(
              states=self.states_spec.signature(batched=True),
              horizons=TensorSpec(type="int", shape=(2,)).signature(
                  batched=True
              ),
              internals=self.internals_spec.signature(batched=True),
              auxiliaries=self.auxiliaries_spec.signature(batched=True),
              actions=self.actions_spec.signature(batched=True),
              reward=self.reward_spec.signature(batched=True),
              reference=self.reference_spec().signature(batched=True),
          )
  
      elif function == "reference":
          return SignatureDict(
              states=self.states_spec.signature(batched=True),
              horizons=TensorSpec(type="int", shape=(2,)).signature(
                  batched=True
              ),
              internals=self.internals_spec.signature(batched=True),
              auxiliaries=self.auxiliaries_spec.signature(batched=True),
              actions=self.actions_spec.signature(batched=True),
          )
  
      else:
          return super().input_signature(function=function)
  
file_path: tensorforce/tensorforce/core/objectives/objective.py
imports: from tensorforce.core import Module, SignatureDict, TensorSpec, tf_function
label_desc: Use image node_id 5 for calling the Objective obj's underlying member method code with example usage: obj.input_signature() and returns: SignatureDict, SignatureDict, super