tf.keras.layers.InputLayer

tf.keras.layers.InputLayer는 신경망의 입력으로 사용되는 층입니다.



예제1

import tensorflow as tf
import numpy as np

# Explicit InputLayer
model = tf.keras.Sequential([
    tf.keras.layers.InputLayer(input_shape=(4,)),
    tf.keras.layers.Dense(8)])

model.compile(tf.optimizers.RMSprop(0.001), loss='mse')
model.fit(np.zeros((10, 4)), np.ones((10, 8)))

tf.keras.layers.InputLayer는 신경망의 입력층입니다.

input_shape을 사용해서 입력의 형태를 지정합니다.



예제2

import tensorflow as tf
import numpy as np

model = tf.keras.Sequential([
    tf.keras.layers.Dense(8, input_shape=(4,))])

model.compile(tf.optimizers.RMSprop(0.001), loss='mse')
model.fit(np.zeros((10, 4)), np.ones((10, 8)))

InputLayer를 사용하지 않고, 첫번째 층에 input_shape을 사용해서 입력의 형태를 지정할 수도 있습니다.

예제1과 예제2의 model은 동일한 신경망 모델입니다.



이전글/다음글


공유하기