jax.scipy.stats.betabinom.pmf

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

Beta-binomial 概率质量函数。

scipy.stats.betabinom pmf 的 JAX 实现。

beta-binomial 分布的概率质量函数定义为

\[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()