code: def kl_divergence_with_logit(q_logit, p_logit):
      """Computes KL-divergence between to sets of logits."""
      q = tf.nn.softmax(q_logit)
      qlogq = -tf.nn.softmax_cross_entropy_with_logits_v2(
          labels=q, logits=q_logit
      )
      qlogp = -tf.nn.softmax_cross_entropy_with_logits_v2(
          labels=q, logits=p_logit
      )
      return qlogq - qlogp
  
imports: import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: kl_divergence_with_logit(q_logit, p_logit) and returns: unknown