I am reading the documentation here for the MPNN. It says that the edge features should be of shape (N, *) where * is the input dim to the function used on the edge features. Previously it is said that node features should also be of shape (N, D_in), so I assumed that N would be the number of nodes, but that wouldn’t make sense for the shape of the edge features. Is the N in the edge features shape the number of edges we have?
Also, how do I set up my edge feature tensor? what do I want the rows to be? I’m assuming order matters somewhere e.g. the first row should be for the edge feature of the node 0 to its first neighbour? is there documentation for this anywhere?