jax.scipy.stats.geom.pmf

jax.scipy.stats.geom.pmf(k, p, loc=0)

Geometric probability mass function.

JAX implementation of scipy.stats.geom pmf.

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:

• k (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, value at which to evaluate the PMF

• p (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, distribution shape parameter

• loc (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, distribution offset parameter

Returns:

array of pmf values.

Return type:

Array

See also

jax.scipy.stats.geom.logpmf()