tensorflow_probability/python/mcmc/hmc.py
def _log_sum_sq(x, axis=None):
  """Computes log(sum(x**2))."""
  return tf.reduce_logsumexp(
      input_tensor=2. * tf.math.log(tf.abs(x)), axis=axis)
def _log_sum_sq(x, axis=None):
  """Computes log(sum(x**2))."""
  return tf.reduce_logsumexp(
      input_tensor=2. * tf.math.log(tf.abs(x)), axis=axis)
def _log_sum_sq ( x , axis = None ) : return tf . reduce_logsumexp ( input_tensor = 2. * tf . math . log ( tf . abs ( x ) ) , axis = axis )
Computes log(sum(x**2)).
Computes log ( sum ( x ** 2 )) .