code: def to_bert_input(token_idx, null_idx):
      """token_idx is a 2D tensor int.
      return token_idx, segment_idx and mask
      """
      segment_idx = token_idx * 0
      mask = token_idx != null_idx
      # nullify elements in case self.NULL_IDX was not 0
      token_idx = token_idx * mask.long()
      return token_idx, segment_idx, mask
  
imports: import os
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from tqdm import tqdm
  from pytorch_transformers.modeling_bert import BertPreTrainedModel, BertConfig, BertModel
  from pytorch_transformers.tokenization_bert import BertTokenizer
  from blink.common.ranker_base import BertEncoder, get_model_obj
  from blink.common.optimizer import get_bert_optimizer
label_desc: Use image node_id 2 for calling a global function with example usage: to_bert_input(token_idx, null_idx) and returns: token_idx,  segment_idx,  mask