Minor fix for FlattenerLayer

Input for raw_layer.get_layer_monitoring_channels should be formatted in the shape of raw_layer.output space, instead of flattener_layer.output_space