code: def confusion_matrix(logits, labels, num_class):
      """confusion matrix candies"""
      return tf.confusion_matrix(
          labels=tf.reshape(labels, [-1]),
          predictions=tf.reshape(tf.argmax(logits, -1), [-1]),
          num_classes=num_class,
      )
  
label_desc: Use image node_id 2 for calling a global function with example usage: confusion_matrix(logits, labels, num_class) and returns: tf