tfp.experimental.substrates.jax.math.log_add_exp

View source on GitHub

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

Aliases:

• tfp.experimental.substrates.jax.math.generic.log_add_exp

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.