Using RGCNConv with heterogeneous graphs (different node types)

[waterhall]

Hello, I'm looking for a way to use RGCNConv with heterogeneous graphs (different node types and edges). I understand that RGCNConv can handle different edge types by passing in edge_type along with the edge_index and x.

How can I modify RGCNConv to take into account the different node types? Or is there another existing method that can handle different node types as well as edge types?

Thank you.

[waterhall]

I could concatenate a one-hot vector indicating the node type to my current node representation, but I was hoping for a way for the RGCN to have nodetype-specific transformations, similar to how it has different weights for each relation type..

[rusty1s]

See #2940 (comment)