In Tensorflow, come svelare gli indici appiattiti ottenuti da tf.nn.max_pool_with_argmax?

Question

incontro un problema: Dopo uso la tf.nn.max_pool_with_argmax, ottengo gli indici cioè argmax: A Tensor of type Targmax. 4-D. The flattened indices of the max values chosen for each output.

Come svelare i indici appiattiti torna all'elenco coordinate in tensorflow?

Grazie mille.

Answer 1

Ho avuto lo stesso problema di oggi e ho finito con questa soluzione:

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) 

Ecco una usage example in un taccuino ipython (lo uso per trasmettere le posizioni di messa in comune argmax al mio metodo unpooling)