Как сделать глобальный средний пул в TensorFlow?

Question

Как сделать глобальный средний пул в TensorFlow? Если у меня есть тензор формы batch_size, height, width, channels = 32, 11, 40, 100, достаточно ли просто использовать tf.layers.average_pooling2d(x, [11, 40], [11, 40]), если каналы = классы?

Answer 1

Вам не нужно проходить ни одного шага. Если padding = 'valid' (по умолчанию), если пространственный масштаб фильтра объединения совпадает с пространственной протяженностью изображения, вы получите изображение 1x1. Кроме того, так вы это делаете, да.

Answer 2

Вы также можете сделать tf.reduce_mean (x, [1,2]), особенно если ваши высота и ширина не определены.