code: def get_axis(axarr, H, W, i, j):
      H, W = H - 1, W - 1
      if not (H or W):
          ax = axarr
      elif not (H and W):
          ax = axarr[max(i, j)]
      else:
          ax = axarr[i][j]
      return ax
  
file_path: robustness/robustness/tools/vis_tools.py
imports: import numpy
  import matplotlib.pyplot
  from sklearn.decomposition import PCA
  from sklearn import manifold
  import seaborn
label_desc: Use image node_id 1 for calling a global function with example usage: get_axis(axarr, H, W, i, j) and returns: ax