code: def axial_2d_pattern(H, W):
      return axial_nd_pattern(H, W)
  
imports: import math
  from typing import List
  import numpy
  import torch
  from xformers.components.attention.sparsity_config import BigBirdSparsityConfig, BSLongformerSparsityConfig, FixedSparsityConfig, VariableSparsityConfig
label_desc: Use image node_id 16 for calling a global function with example usage: axial_2d_pattern(H, W) and returns: axial_nd_pattern