self.flatten in graph_signal_classification_mnist.py

[yhh]

It appears that replacing self.flatten with Flatten() could easily boost the performance.

[danielegrattarola]

That is likely, but it is common in graph machine learning to use permutation invariant readout functions (like sum or average). What kind of performance improvement did you have?