TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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.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.