code: def _get_token_logits(self, logits, pad_mask):
      pad_start = pad_mask.count(self.vocab.get_pad_index())
      return logits[:pad_start]
  
imports: import itertools
  from typing import List
  import torch
  from pytext.data.bert_tensorizer import BERTTensorizer
  from pytext.data.roberta_tensorizer import RoBERTaTensorizer
  from pytext.data.tensorizers import lookup_tokens
  from pytext.data.utils import pad_and_tensorize
  from pytext.torchscript.tensorizer import ScriptRoBERTaTensorizerWithIndices
  from pytext.torchscript.vocab import ScriptVocabulary
label_desc: Use image node_id 6 for calling the SquadForRoBERTaTensorizerForKD obj's underlying member method code with example usage: obj._get_token_logits(logits, pad_mask) and returns: logits