[TensorFlow] Reshape 레이어 예제
tf.keras.layers.Reshape
레이어는 입력 데이터의 형태를 변경하는 역할을 합니다. 아래는 tf.keras.layers.Reshape
레이어를 사용한 예제입니다.
import tensorflow as tf
## 더미 데이터 생성
dummy_data = tf.random.normal(shape=(4, 16)) # 4개의 샘플, 각 샘플은 16개의 특성으로 구성
## Reshape 레이어 정의
reshape_layer = tf.keras.layers.Reshape(target_shape=(8, 2)) # 8x2 형태로 변환
## Reshape 레이어 실행
reshaped_data = reshape_layer(dummy_data)
print("Input shape:", dummy_data.shape)
print("Reshaped data shape:", reshaped_data.shape)`
위 예제에서는 tf.keras.layers.Reshape
레이어를 사용하여 4x16 형태의 더미 데이터에 대한 레이어를 정의하고 실행합니다. target_shape
매개변수로 목표 형태를 지정하여 데이터의 형태를 변경합니다.
실행 결과 예시:
Input shape: (4, 16)
Reshaped data shape: (4, 8, 2)`
위에서 확인할 수 있듯이, 입력 데이터의 형태가 (4, 16)에서 (4, 8, 2)로 변경되었습니다. tf.keras.layers.Reshape
레이어를 사용하여 입력 데이터의 형태를 조정하여 다른 레이어에 입력으로 전달할 수 있습니다.