如果我有一个 4-D blob，比如大小为 (40,1024,300,1(，我想在第二个通道上平均池并生成大小为 (40,1,300,1( 的输出，我将如何做？我认为缩减层会折叠整个 blob 并通过对所有其他轴(1 之后(中的元素求和来生成大小为 (40( 的 blob。在不重新实现新层的情况下，是否有任何解决方法？
我发现的唯一简单的解决方法如下。将 blob 排列为形状(40,300,1,1024)。使用约简层计算具有axis = -1和operation = MEAN的平均值。我认为斑点的形状将是(40,300,1).您可能需要使用 reshape 在末尾附加一个额外的维度(检查是否需要这样做(，然后排列回形状(40,1,300,1) 。
您可以在此处或此处找到Permute层的实现。我希望这有所帮助。