多層損失tensorflow

Question

我想建立我使用的是多層損失 AlexNet作爲我的基礎網絡，我有4類，有3個可能的標籤，所以我試圖建立它如下：

output_gt = tf.placeholder(tf.int32, [None,4,3], name='output') 

這不是我的真實輸出，但這是他的大小，這意味着alexNet上的輸出層的大小爲[4,3] 我希望能夠僅查看與我放置的類相關的輸出input.so最後爲每個圖像我會得到[1,3]尺寸輸出，從相關部分來的原始輸出 例如

batch_size =2 
labels = [0,2] 
output = [*batc_size_dim*][[0 0 0], [1 1 1],[2 2 2 ],[3 3 3]] 

我會得到

我如何使用標籤和輸出得到new_output 我嘗試使用面具和我沒有 你能幫助我嗎？

Answer 1

請嘗試以下操作： tf.gather(output, input)