(logits, reduce_axis=-1)
return logits - tf.reduce_logsumexp(logits, axis=reduce_axis, keepdims=True)
return log prob use log sum func
return log prob use log sum func
return log prob use log sum func
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)
def log_prob_from_logits ( logits , reduce_axis = - 1 ) : return logits - tf . reduce_logsumexp ( logits , axis = reduce_axis , keepdims = True )