code: def _test_local_mask():
      """
      Test local mask
      """
      from labml.logger import inspect
  
      inspect(AFTLocal.create_local_mask(10, 4))
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/transformers/aft/__init__.py
imports: from typing import Optional
  import torch
  from torch import nn
  from labml_helpers.module import Module
label_desc: Use image node_id 1 for calling a global function with example usage: _test_local_mask() without return types