code: def get_mask(lengths: torch.Tensor):
      max_seq_len = lengths.max()
      att_mask = torch.ones(
          1, max_seq_len, max_seq_len, dtype=torch.bool
      )
  
      pad_mask, att_mask = _create_masks(att_mask, max_seq_len, lengths)
      return pad_mask, att_mask
  
imports: import pytest
  import torch
  from nemo.collections.asr.parts.submodules import adapters
  from nemo.core.classes.mixins import adapter_mixin_strategies
  from nemo.utils import config_utils
label_desc: Use image node_id 2 for calling a global function with example usage: get_mask(lengths) and returns: pad_mask,  att_mask