code: def attention_mask_func(attention_scores, attention_mask):
      return attention_scores.masked_fill(attention_mask, -10000.0)
  
imports: import itertools
  import torch
  from torch.testing._internal import common_utils
  from apex.transformer import AttnMaskType
  from apex.transformer.functional import FusedScaleMaskSoftmax
label_desc: Use image node_id 1 for calling a global function with example usage: attention_mask_func(attention_scores, attention_mask) and returns: attention_scores