logaddexp2

Contents

logaddexp2#

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

Logarithm of the sum of exponentiations of the inputs in base 2.

Computes log2(2**x + 2**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 log2(2**x + 2**y).

Return type:

Array

Examples

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