在caffe中的通道中求和元素

Question

如果我有一个4-D blob，比如大小（40,1024,300,1），我想要在第二个通道上平均池并生成一个大小为（40,1,300,1）的输出，怎么会我做到了吗？我认为缩小层会折叠整个blob并通过对所有其他轴（1之后）中的元素求和来生成大小（40）的blob。没有重新实现新图层，有没有解决这个问题？

Answer 1

我发现的唯一简单的解决方法如下。将blob置换为形状(40,300,1,1024)。使用缩小图层计算axis = -1和operation = MEAN的均值。我认为blob的形状为(40,300,1)。您可能需要使用reshape在末尾添加额外的维度（检查是否需要），然后置换回(40,1,300,1)形状。

您可以找到Permute图层here或here的实现。我希望这会有所帮助。