logaddexp

Contents

logaddexp#

class saiunit.math.logaddexp(x, y, unit_to_scale=None, **kwargs)#

Logarithm of the sum of exponentiations of the inputs.

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

Parameters:

  • x (Union[Array, ndarray, number, bool, saiunit.Quantity]) – First input.

  • y (Union[Array, ndarray, number, bool, saiunit.Quantity]) – Second input. Must be broadcastable with x.

  • unit_to_scale (Optional[saiunit.Unit]) – Unit used to convert both inputs to dimensionless numbers.

Returns:

out – Element-wise log(exp(x) + exp(y)).

Return type:

Array

Examples

>>> import saiunit as u
>>> import jax.numpy as jnp
>>> u.math.logaddexp(jnp.array([1.0]), jnp.array([2.0]))
Array([2.3132617], dtype=float32)
Contents