how flatten works code example

Example 1: flatten tf keras

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

Example 2: how to flatten the image dataset

# Reshape the training and test examples 
train_x_flatten = train_x_orig.reshape(train_x_orig.shape[0], -1).T   # The "-1" makes reshape flatten the remaining dimensions
test_x_flatten = test_x_orig.reshape(test_x_orig.shape[0], -1).T