code: def b_mas(b_log_attn_map, in_lens, out_lens, width=1):
      assert width == 1
      attn_out = np.zeros_like(b_log_attn_map)
  
      for b in prange(b_log_attn_map.shape[0]):
          out = mas_width1(
              b_log_attn_map[b, 0, : out_lens[b], : in_lens[b]]
          )
          attn_out[b, 0, : out_lens[b], : in_lens[b]] = out
      return attn_out
  
imports: from enum import Enum
  from typing import Optional, Tuple
  import librosa
  import matplotlib.pylab
  import numpy
  import torch
  from einops import rearrange
  from numba import jit, prange
  from nemo.collections.tts.torch.tts_data_types import DATA_STR2DATA_CLASS, MAIN_DATA_TYPES, WithLens
  from nemo.utils import logging
  from nemo.utils.decorators import deprecated
label_desc: Use image node_id 10 for calling a global function with example usage: b_mas(b_log_attn_map, in_lens, out_lens, width) and returns: attn_out