我浏览了tf.nn.max_pool_with_argmax的文档
对输入执行最大池化并输出最大值和索引。
argmax 中的索引被展平,因此位置 [b, y, x, c] 处的最大值变为展平索引 ((b * height + y) * width + x) * channels + c。
返回的索引在展平之前总是在 [0, height) x [0, width) 中,即使涉及填充并且数学上正确的答案在外面(负数或太大)。这是一个错误,但很难以安全的向后兼容方式修复它,尤其是由于扁平化。
变量 b、y、x 和 c 尚未明确定义,因此我在实现此方法时遇到问题。有人可以提供相同的吗。