W Tensorflow, jak rozwikłać spłaszczone indeksy uzyskane przez tf.nn.max_pool_with_argmax?

Question

spotykam problem: Po używam tf.nn.max_pool_with_argmax, że uzyskanie wskaźników tj argmax: A Tensor of type Targmax. 4-D. The flattened indices of the max values chosen for each output.

Jak rozwikłać spłaszczone indeksy z powrotem do listy Współrzędne w Tensorflow?

Dziękuję bardzo.

Answer 1

miałem ten sam problem dzisiaj i skończyło się z tego rozwiązania:

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) 

Oto usage example w notatniku ipython (używam go do przekazania stanowiska pooling argmax do mojego unpooling metody)