En Tensorflow, ¿cómo desentrañar los índices aplanados obtenidos por tf.nn.max_pool_with_argmax?

Me encuentro con un problema: después de usar tf.nn.max_pool_with_argmax , obtengo los índices, es decir, argmax: A Tensor of type Targmax. 4-D. The flattened indices of the max values chosen for each output. argmax: A Tensor of type Targmax. 4-D. The flattened indices of the max values chosen for each output.

¿Cómo desentrañar los índices aplanados de nuevo a la lista de coordenadas en Tensorflow?

Muchas gracias.

Hoy tuve el mismo problema y terminé con esta solución:

 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) 

Aquí hay un ejemplo de uso en un cuaderno ipython (lo uso para enviar las posiciones de agrupación argmax a mi método de desagrupación)