jax.scipy.stats.geom.logpmf

Contents

jax.scipy.stats.geom.logpmf#

jax.scipy.stats.geom.logpmf(k, p, loc=0)[source]#

Geometric log probability mass function.

JAX implementation of scipy.stats.geom logpmf.

The Geometric probability mass function is given by

\[f(k) = (1 - p)^{k-1}p\]

for \(k\ge 1\) and \(0 \le p \le 1\).

Parameters:
Returns:

array of logpmf values.

Return type:

Array