Closed tclements closed 2 years ago
Fortunately, Flux.outputsize
implementation is so generic that it basically works with anything, you just need to wrap into closures in some cases:
julia> Flux.outputsize(wg, (3, 10))
(2, 10)
julia> Flux.outputsize(x -> model(g, x), (3, 10))
(2, 10)
Great, thank you!
This is a feature request: it'd be nice to have extend the functionality of
Flux.outputsize
toGNNChain
s. I imagine this could be applied to either aWithGraph
or aGNNGraph
and Tuple ofinputsize
. Here's a sketch of a MWE from the docs: