jax.scipy.stats.geom.pmf#
- jax.scipy.stats.geom.pmf(k, p, loc=0)[源码]#
几何分布的概率质量函数。
JAX 对
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 | TypedNdArray) – 数组或类数组，用于评估 PMF 的值
p (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – 数组或类数组，分布的形状参数
loc (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – 数组或类数组，分布的偏移参数
- 返回:
pmf 值的数组。
- 返回类型: