Trong Tensorflow, làm thế nào để làm sáng tỏ các chỉ số san phẳng thu được bằng tf.nn.max_pool_with_argmax?

Question

tôi gặp một vấn đề: Sau khi tôi sử dụng tf.nn.max_pool_with_argmax, tôi có được các chỉ số ví dụ: argmax: A Tensor of type Targmax. 4-D. The flattened indices of the max values chosen for each output.

Làm thế nào để làm sáng tỏ những chỉ số phẳng trở lại danh sách tọa độ trong Tensorflow?

Cảm ơn bạn rất nhiều.

Answer 1

tôi đã cùng một vấn đề ngày hôm nay và tôi đã kết thúc với giải pháp này:

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) 

Đây là một usage example trong một máy tính xách tay ipython (tôi sử dụng nó để chuyển tiếp các vị trí tổng hợp argmax phương pháp unpooling của tôi)