class_docstr: A simplified PNAConv variant used in OGB submissions
class_docstr_tok: ["A","simplified","PNAConv","variant","used","in","OGB","submissions"]
code: def reduce(self, nodes):
      h = nodes.mailbox["m"]
      D = h.shape[-2]
      h = torch.cat(
          [aggregate(h) for aggregate in self.aggregators], dim=1
      )
      h = torch.cat(
          [scale(h, D=D, delta=self.delta) for scale in self.scalers],
          dim=1,
      )
      return {"h": h}
  
imports: from typing import List
  import dgl.function
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from dgl.nn import AvgPooling, SumPooling
  from ogb.graphproppred.mol_encoder import AtomEncoder
label_desc: Use image node_id 2 for calling the SimplePNAConv obj's underlying member method code with example usage: obj.reduce(nodes) and returns: dict