张量流中的求和池

Question

张量流中是否有现有的总和池实现？在搜索文档时，似乎只支持平均和最大池操作。

我有一个 NHWC 张量，其中每个 HxW 矩阵都是一个概率图，我想通过 2x2 总和池对其进行下采样。如果张量流中不存在求和池函数，是否有其他方法可以使用 Python API 实现此下采样操作？

Answer 1

目前tensorflow中还没有这样的东西。下一个最好的办法可能是将平均池化的输出乘以 4。逐元素乘法足够快，无需担心这个额外的步骤。

不过，在这个过程中你确实会失去一些精度。

另一种方法是自己计算总和，例如

y = (x[:, 0::2, 0::2] +
     x[:, 1::2, 0::2] +
     x[:, 0::2, 1::2] +
     x[:, 1::2, 1::2])

它应该比以前的解决方案慢，但精度可能会稍微好一点，如果这对您很重要的话。