多层损失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)