I am at a loss about the following. I am building a small 2D network where the last 2 layers are a
MaxPool followed by a
Dense. The parameters to create
MaxPool do not care about the size of the input. But the
Dense just afterwards do.
I have tried many ways to extract this work but without success. (A
Flux.flatten is in between) . Hoping for guidance elsewhere, I have gone through the source code of Flux and NNLib. Every single example hard-codes the input size of
Dense or cheats with keeping the stride of the
MaxPool at 1.
I am sure I am miising something stupid.