code: def type_inference(self):
      reduced_shape = self._find_reduced_shape()
      return types.tensor(types.int32, tuple(reduced_shape))
  
imports: import numpy
  from coremltools.converters.mil.mil import Operation, precondition, types
  from coremltools.converters.mil.mil.input_type import DefaultInputs, InputSpec, TensorInputType
  from coremltools.converters.mil.mil.operation import VALUE
  from coremltools.converters.mil.mil.ops.defs._op_reqs import register_op
  from coremltools.converters.mil.mil.types import nptype_from_builtin
label_desc: Use image node_id 2 for calling the reduce_arg obj's underlying member method code with example usage: obj.type_inference() and returns: types