megatron.model.utils.attention\_mask\_func
==========================================

.. currentmodule:: megatron.model.utils

.. autofunction:: attention_mask_func