(axis=-1)
return Fn('LogSumExp',
            lambda x: fastmath.logsumexp(x, axis=axis, keepdims=True))
Returns a layer that computes log(sum(exp(x))) along one tensor axis.

  Args:
    axis: Axis along which values are grouped for computing log-sum-exp.
Returns a layer that computes log(sum(exp(x))) along one tensor axis.
Returns a layer that computes log ( sum ( exp ( x ))) along one tensor axis .
def LogSumExp(axis=-1):
  """Returns a layer that computes log(sum(exp(x))) along one tensor axis.

  Args:
    axis: Axis along which values are grouped for computing log-sum-exp.
  """
  return Fn('LogSumExp',
            lambda x: fastmath.logsumexp(x, axis=axis, keepdims=True))
def LogSumExp ( axis = - 1 ) : return Fn ( 'LogSumExp' , lambda x : fastmath . logsumexp ( x , axis = axis , keepdims = True ) )