how to flatten select dimensions in keras code example

Example: flatten tf keras

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