code: def mask_padded_tokens(tokens, pad_id):
      mask = tokens != pad_id
      return mask
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 3 for calling a global function with example usage: mask_padded_tokens(tokens, pad_id) and returns: mask