是否有一种有效的方法在keras中应用径向平均值？

Question

我想在keras管道的末端应用径向平均值。

在倒数第二个步骤中，我得到的图像大小为n x n。然后，我想将此n x n图像映射到1 x n / 2向量，其中vector [x] = mean（image（radialPosition = x））。即我想对距图像中心的距离X的所有点求平均，并将其设置为output [x]。我们可以假设n为奇数，因此中心点为单点。

我已经考虑过遍历所有半径，并选择所需的索引，以及图像与多个“平均”矩阵之间的点积，但这些方法似乎都没有计算效率。

有更好的方法吗？