jax.scipy.stats.geom.pmf

jax.scipy.stats.geom.pmf#

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

几何概率质量函数。

JAX implementation of scipy.stats.geom pmf.

几何概率质量函数由以下公式给出

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

对于 \(k\ge 1\) 和 \(0 \le p \le 1\)。

参数:

k (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 用于评估PMF的值
p (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 分布形状参数
loc (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 分布偏移参数

返回:

pmf 值的数组。

返回类型:

Array

参见