jax.scipy.stats.nbinom.logpmf

jax.scipy.stats.nbinom.logpmf(k, n, p, loc=0)

负二项分布对数概率质量函数。

JAX implementation of 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) – arraylike, 用于评估PMF的值

• n (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 分布形状参数

• p (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 分布形状参数

• loc (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 分布偏移参数

返回:

对数概率密度值的数组。

返回类型:

Array

参见

jax.scipy.stats.nbinom.pmf()