jax.scipy.stats.betabinom.pmf

jax.scipy.stats.betabinom.pmf(k, n, a, b, loc=0)

Beta-二项概率质量函数。

scipy.stats.betabinom pmf 的 JAX 实现。

Beta-二项分布的概率质量函数定义为

\[f(k, n, a, b) = {n \choose k}\frac{B(k+a,n-k-b)}{B(a,b)}\]

其中 \(B(a, b)\) 是 beta() 函数。它为 \(n\ge 0\)，\(a>0\)，\(b>0\) 和非负整数 k 定义。

参数:

• k (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，用于评估 PMF 的值

• n (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布的形状参数

• a (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布的形状参数

• b (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布的形状参数

• loc (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布的偏移参数

返回:

pmf 值的数组

返回类型:

Array

另请参阅

jax.scipy.stats.betabinom.logpmf()