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.