code: def log_prob_from_logits(logits, reduce_axis=-1):
      """return log prob use log sum func"""
      return logits - tf.reduce_logsumexp(
          logits, axis=reduce_axis, keepdims=True
      )
  
file_path: delta/delta/layers/utils_tf.py
imports: import math
  import numpy
  import tensorflow
  from tensorflow.python.util import nest
label_desc: Use image node_id 1 for calling a global function with example usage: log_prob_from_logits(logits, reduce_axis) and returns: unknown