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