Set self.input_space for FlattenerLayer too.

self.input_space wasn't set up for FlattenerLayer.