tf.keras.layers.flatten how to set the input_shape code example

Example: flatten tf keras

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