jax.scipy.stats.geom.pmf

jax.scipy.stats.geom.pmf#

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

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