Keras中的2D局部最大搜索和求和层

Question

我需要Keras中的一个（不可训练的）图层，它对2-dim输入进行局部最大搜索，并在该最大值附近的某个环境中汇总所有条目。图层的输入形状类似于（无，1，24,16）。

为了更清楚，让我们看一下输入的样子：

该层的所需输出应为：

因此它应找到每个局部最大值，然后将（最多）8个相邻像素中的所有条目总和到该最大值。

这是一个高能物理相关的问题，所以我认为对这些“图像”的解释不应该太重要。

如何使用Keras图层实现这一目标？

有什么建议吗？谢谢！