self_loop_edges = torch.zeros(x.size(0), edge_attr.size(1)).to(edge_index.device) # new edge_attr = torch.cat([edge_attr, self_loop_edges], dim=0) # new 3. Start propagating messages self.propagate(edge_index, x=x, edge_attr=edge_attr, size=size) # new # 2 x (E+N), ...