code: def __init__(
      self,
      start_logits_column=Config.start_logits_column,
      end_logits_column=Config.end_logits_column,
      has_answer_logits_column=Config.has_answer_logits_column,
      pad_mask_column=Config.pad_mask_column,
      segment_labels_column=Config.segment_labels_column,
      **kwargs,
  ):
      super().__init__(**kwargs)
      self.start_logits_column = start_logits_column
      self.end_logits_column = end_logits_column
      self.has_answer_logits_column = has_answer_logits_column
      self.pad_mask_column = pad_mask_column
      self.segment_labels_column = segment_labels_column
  
imports: from typing import List
  import torch
  from pytext.common.constants import SpecialTokens
  from pytext.config.component import ComponentType, create_component
  from pytext.data.tensorizers import TokenTensorizer
  from pytext.data.tokenizers import Tokenizer, WordPieceTokenizer
  from pytext.data.utils import pad_and_tensorize, VocabBuilder, Vocabulary
label_desc: Use image node_id 2 to create a new SquadTensorizerForKD object from inherited base classes: SquadTensorizer with example: obj = SquadTensorizerForKD(start_logits_column, end_logits_column, has_answer_logits_column, pad_mask_column, segment_labels_column)