Flatten TF romaty

tf.keras.layers.Flatten(
    data_format=None, **kwargs
)
Niranth_S