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]）を指定することもできます。