pyg-team/pytorch_geometric - github.com
github.com › pyg-team › pytorch_geometricNov 20, 2019 · Following up on this, updating edge features is quite trivial without any helper functions, e.g.: row, col = edge_index new_edge_attr = self. mlp ( torch. cat ( [ x [ row ], x [ col ], edge_attr ], dim=-1 )) Providing a more elegant API in analogy to MessagePassing is interesting though. Will think about it!