It would be nice to visualize the final layer of a CNN being flattened when entering into a FeedForwardLayer.