jax.scipy.stats.geom.pmf¶

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

Probability mass function at k of the given RV.

LAX-backend implementation of pmf().

Original docstring below.

Parameters

k (array_like) – Quantiles.
arg1 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
arg2 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
arg3 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
... (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
loc (array_like, optional) – Location parameter (default=0).

Returns

pmf – Probability mass function evaluated at k

Return type

array_like