FluxML / GeometricFlux.jl

Geometric Deep Learning for Flux
https://fluxml.ai/GeometricFlux.jl/stable/
MIT License
348 stars 30 forks source link

GATConv aggregates by default #129

Closed ilancoulon closed 3 years ago

ilancoulon commented 3 years ago

GATConv seems to aggregate the features by default.

So if I give a FxN matrix to a GATConv, the output is a F' vector (where N is the number of nodes in the graph, and F, F' are the dimensions of the features.

Why is that? Is it on purpose?

Because it makes it impossible to have a chain of GATConv for example.

ilancoulon commented 3 years ago

I had not really understood what the aggregation was, my bad.