loss1 = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(prediction1,
    labels_1))
loss2 = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(prediction2,
    labels_2))
loss = loss1 + loss2
Многоуровневый классификатор в Tensorflow