code: def _get_doc_logits(self, logits, pad_mask, segment_labels):
      ques_seq_len = segment_labels.index(1)
      try:
          pad_start = pad_mask.index(0)
      except ValueError:  # 0 doesn't exits in pad_mask
          pad_start = len(logits)
      return logits[ques_seq_len : pad_start - 1]
  
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 5 for calling the SquadTensorizerForKD obj's underlying member method code with example usage: obj._get_doc_logits(logits, pad_mask, segment_labels) and returns: logits