code: def _reduce_shape_func(data_shape, axis_record):
      out = output_tensor((len(axis_record),), "int64")
      for i in const_range(len(axis_record)):
          if axis_record[i] >= 0:
              out[i] = data_shape[axis_record[i]]
          else:
              out[i] = int64(1)
  
      return out
  
file_path: tvm/python/tvm/relay/op/_reduce.py
imports: from __future__ import absolute_import
  from tvm.runtime import convert
  from tvm.te.hybrid import script
  from tvm.topi.utils import get_const_int, get_const_tuple
  from .None import op
label_desc: Use image node_id 2 for calling a global function with example usage: _reduce_shape_func(data_shape, axis_record) and returns: out