class_docstr: Classifier head for Vision Transformer
  
  Args:
  ----
      num_classes: number of output classes
class_docstr_tok: ["Classifier","head","for","Vision","Transformer","Args",":","--","--","num_classes",":","number","of","output","classes"]
code: def __init__(self, num_classes: int) -> None:
      super().__init__()
  
      self.head = layers.Dense(
          num_classes, kernel_initializer="he_normal", name="dense"
      )
  
file_path: doctr/doctr/models/classification/vit/tensorflow.py
imports: from copy import deepcopy
  from typing import Any, Dict, Optional, Tuple
  import tensorflow
  from tensorflow.keras import Sequential, layers
  from doctr.datasets import VOCABS
  from doctr.models.modules.transformer import EncoderBlock
  from doctr.models.modules.vision_transformer.tensorflow import PatchEmbedding
  from doctr.utils.repr import NestedObject
  from ...utils import load_pretrained_params
label_desc: Use image node_id 1 to create a new ClassifierHead object from inherited base classes: layers, NestedObject with example: obj = ClassifierHead(num_classes)