jax.scipy.stats.nbinom.logpmf

内容

jax.scipy.stats.nbinom.logpmf(k, n, p, loc=0)[source]#

负二项式对数概率质量函数。

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) – 数组类，分布偏移参数

返回值:

对数概率密度值数组。

返回类型:

数组

参见