“Cómo aplanar el conjunto de datos de imagen” Código de respuesta

Cómo aplanar el conjunto de datos de imagen

# 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