class_docstr: output logits 
class_docstr_tok: ["output","logits"]
code: def __init__(self, num_class):
      super().__init__(name="head")
      self.num_class = num_class
  
      self.fc = tf.keras.layers.Dense(self.num_class)
  
imports: import numpy
  from absl import logging
  import delta.compat
  from tensorflow.python.keras import backend
  from delta.models.base_model import Model
  from delta.layers.base_layer import Layer
  from delta import utils
  from delta.utils.register import registers
label_desc: Use image node_id 1 to create a new Head object from inherited base classes: Layer with example: obj = Head(num_class)