Flatten 라이브러리 사용하는 이유

model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(128, 'relu'))
model.add(tf.keras.layers.Dense(10, 'softmax'))
model.compile('adam', loss = 'sparse_categorical_crossentropy', metrics = ['accuracy'])
model.fit(training_images, training_labels, epochs = 5)

 

 

Fashion Mnist 이미지 데이터를 분류할 모델이다.이미지는 28 * 28인데 이것을 Flatten으로 입력받아서 딥러닝을 수행한다.