jax_privacy.matrix_factorization.buffered_toeplitz.geometric_sum

jax_privacy.matrix_factorization.buffered_toeplitz.geometric_sum(a, r, num=inf)[source]

Sum a + a*r + a*r**2 + … + a*r**(num-1) (or limit if num=jnp.inf).

Parameters:

a (Array) – Scale factor (or vector of scale factors).
r (Array) – ratio between successive terms, requires \(|r| < 1\).
num (Union[Array, ndarray, bool, number, float, int]) – How many terms to add, or jnp.inf for the limit.

Return type:

Array

Returns:

The sum.