Альтернатива глобальному пулу с использованием tf.reduce_mean?

Question

output = tf.reduce_mean(net, [1, 2], name='global_pool', keep_dims=False) 

net имеет форму (размер пакета, высота, ширина, # каналы)

Есть ли другой способ сделать глобальный пул и получить тот же результат?

Answer 1

У меня есть вычислительная система, очень сложная, но она работает:

1. Использование tf.split(..., axis=-1)
2. Выполните tf.layers.average_pooling2d для каждого разбиения и выберите размер ядра = тензорный размер

Второй вариант: tf.keras.layers.GlobalAveragePooling2D, но я бы предложил использовать tf.reduce_mean под капотом, но dint посмотрел в источнике.