code: def DynamicLazyAxis(shape, idx):
      return lambda: shape[idx]
  
file_path: tensorpack/tensorpack/models/shape_utils.py
imports: import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: DynamicLazyAxis(shape, idx) and returns: function