megatron.model.utils.attention_mask_func#

megatron.model.utils.attention_mask_func(attention_scores, attention_mask)#