code: def log_prob_from_logits(logits, reduce_axis=-1):
      return logits - tf.reduce_logsumexp(
          input_tensor=logits, axis=reduce_axis, keepdims=True
      )
  
file_path: finetune/finetune/util/beam_search.py
imports: import math
  import numpy
  from finetune.util.shapes import shape_list
  import tensorflow
  from tensorflow.python.ops import inplace_ops
  from tensorflow.python.util import nest
label_desc: Use image node_id 3 for calling a global function with example usage: log_prob_from_logits(logits, reduce_axis) and returns: unknown