x = layers.Flatten()(x) code example

Example: flatten tf keras

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