Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Basic GGCN edge types support #1854

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

mike-grayhat
Copy link

This is a basic implementation of edge type support for GGCN. At the moment it doesn't support fused operation since message_and_aggregate accepts lifted adjacency matrix, but in case of multiple edge types it should accept list of sparse tensors

    def message_and_aggregate(self, adj_t: SparseTensor, x: Tensor, edge_attr: OptTensor) -> Tensor:
        return matmul(adj_t, x, reduce=self.aggr)

@rusty1s
Copy link
Member

rusty1s commented Nov 25, 2020

Thank you very much! I will have a look ASAP.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants