在Tensorflow中，如何解開由tf.nn.max_pool_with_argmax獲得的扁平索引？

Question

我遇到一個問題：我用的是tf.nn.max_pool_with_argmax後，我獲得了指數即 argmax: A Tensor of type Targmax. 4-D. The flattened indices of the max values chosen for each output.

如何解開扁平指數回在Tensorflow座標列表？

非常感謝。

Answer 1

今天我有同樣的問題，我結束了此解決方案：

def unravel_argmax(argmax, shape): 
    output_list = [] 
    output_list.append(argmax // (shape[2] * shape[3])) 
    output_list.append(argmax % (shape[2] * shape[3]) // shape[3]) 
    return tf.pack(output_list) 

這裏是一個IPython的筆記本usage example（我用它來統籌argmax位置轉發到我的unpooling方法）