code: def data_gen_for_question_answering():
      # LM data gen
      # the `labels` of LM is the token of the output, cause no padding, use `input_ids` as `labels`
      data = data_gen()
      data["start_positions"] = torch.tensor([0])
      data["end_positions"] = torch.tensor([1])
      return data
  
imports: import torch
  import transformers
  from ..registry import ModelAttribute, model_zoo
label_desc: Use image node_id 4 for calling a global function with example usage: data_gen_for_question_answering() and returns: data