code: def global_example_pool(x, batch, size=None):
      size = batch.max().item() + 1 if size is None else size
      return scatter(x, batch, dim=0, dim_size=size, reduce="add")
  
label_desc: Use image node_id 1 for calling a global function with example usage: global_example_pool(x, batch, size) and returns: scatter