I am now considering a heterogeneous graph containing 3 types of nodes, each connected to each other. I consider using the HeteroGraphConv() function to implement message passing, but find that the HeteroGraphConv() function is unable to input edge features.
I want to use EdgeGATConv as a sub-module of HeteroGraphConv(), but I don’t understand the usage of mod_args and mod_kwargs, how can I properly use EdgeGATConv in HeteroGraphConv()?
Or, is there any other way to implement message passing in heterogeneous graphs that takes into account edge features?
Below is my current HeteroGraphConv() construction and I don’t know how to construct the forward function.