code: def _max_reduce_func(inputs, dim):
      return th.max(inputs, dim=dim)[0]
  
imports: from functools import partial
  import torch
  import torch.nn
  from ...base import DGLError
label_desc: Use image node_id 1 for calling a global function with example usage: _max_reduce_func(inputs, dim) and returns: th