code: def attention_mask_func(attention_scores, attention_mask):
      attention_scores.masked_fill_(attention_mask, -10000.0)
      return attention_scores
  
file_path: megatron-lm/megatron/model/utils.py
imports: import math
  import torch
  from megatron import get_args
  from megatron.model import LayerNorm, RMSNorm
label_desc: Use image node_id 3 for calling a global function with example usage: attention_mask_func(attention_scores, attention_mask) and returns: attention_scores