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

Neural Network의 Input으로 사용하기 위하여 행렬이 아닌 배열로 만들어 주어야 한다.

Flatten 라이브러리를 사용하여 2차원 3차원을 1차원으로 바꿔준다.

 

model = tf.keras.models.Sequential()
# layer를 만들기 전에 1차원으로 만들어준다.
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(128, 'relu'))
model.add(tf.keras.layers.Dense(10, 'softmax'))