jax.scipy.stats.nbinom.logpmf#
- jax.scipy.stats.nbinom.logpmf(k, n, p, loc=0)[源代码]#
负二项分布的对数概率质量函数。
JAX实现的
scipy.stats.nbinom
的logpmf
。负二项分布的概率质量函数由下式给出:
\[f(k) = {{k+n-1} \choose {n-1}}p^n(1-p)^k\]对于 \(k \ge 0\) 和 \(0 \le p \le 1\)。
- 参数:
k (Array | ndarray | bool | number | bool | int | float | complex) – 类数组,用于评估PMF的值。
n (Array | ndarray | bool | number | bool | int | float | complex) – 类数组,分布形状参数。
p (Array | ndarray | bool | number | bool | int | float | complex) – 类数组,分布形状参数。
loc (Array | ndarray | bool | number | bool | int | float | complex) – 类数组,分布偏移参数。
- 返回:
对数概率密度函数值的数组。
- 返回类型: