code: def test_vertical_axial_2d_distance(H, W, p):
      d = AP.vertical_axial_2d_distance(H, W, p=p)
      d_ref = _vertical_axial_2d_distance(H, W, p=p)
      assert torch.allclose(d, d_ref)
  
imports: import itertools
  import pytest
  import torch
  import xformers.components.attention.attention_patterns
  from xformers.components.attention.sparsity_config import BigBirdSparsityConfig, BSLongformerSparsityConfig, DenseSparsityConfig, FixedSparsityConfig, VariableSparsityConfig
label_desc: Use image node_id 9 for calling a global function with example usage: test_vertical_axial_2d_distance(H, W, p) without return types