如何在TensorFlow中进行全球平均合并？

Question

如何在TensorFlow中进行全球平均合并？如果我有一个形状的张量,batch_size, height, width, channels = 32, 11, 40, 100只要使用tf.layers.average_pooling2d(x, [11, 40], [11, 40])channels = classes 就足够了吗？

Answer 1

您还可以执行tf.reduce_mean(x,axis = [1,2]),特别是如果未定义高度和宽度.

通常,在CNN中,张量的形状在b, h, w, c哪里b是批量大小,w并且h对应于宽度和高度尺寸,并且c是通道/滤波器的数量.

沿轴[1,2]减小时,减小张量的第一维和第二维(保持批量大小和通道/过滤器的数量)