tfp.experimental.substrates.jax.math.log_add_exp

View source on GitHub

Computes log(exp(x) + exp(y)) in a numerically stable way.

tfp.experimental.substrates.jax.math.log_add_exp(
    x, y, name=None
)

Args:

  • x: float Tensor broadcastable with y.
  • y: float Tensor broadcastable with x.
  • name: Python str name prefixed to Ops created by this function. Default value: None (i.e., 'log_add_exp').

Returns:

  • log_add_exp: log(exp(x) + exp(y)) computed in a numerically stable way.