code: def aggregate_max(h):
      """max aggregation"""
      return torch.max(h, dim=1)[0]
  
imports: import numpy
  import torch
  import torch.nn
label_desc: Use image node_id 2 for calling a global function with example usage: aggregate_max(h) and returns: torch