потери Многослойной 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)