tf.keras.layers.GlobalAveragePooling2D¶

tf.keras.layers.GlobalAveragePooling2D은 Global average pooling 연산을 수행합니다.

예제¶

import tensorflow as tf

input_shape = (2, 4, 5, 3)
x = tf.random.normal(input_shape)
y = tf.keras.layers.GlobalAveragePooling2D()(x)

print(y.shape)

(2, 3)

입력의 형태 (2, 4, 5, 3)는 주로 (batch_size, height, width, depth)를 의미합니다.

Global Average Pooling은 이 입력에 대해 전역 평균을 풀링하는 방식입니다.

출력의 형태가 (batch_size, depth)인 (2, 3)이 됩니다.