jax.scipy.stats.betabinom.pmf

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

Beta-binomial 概率质量函数。

JAX 实现 scipy.stats.betabinom pmf。

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 (数组 | ndarray | 布尔值 | 数字 | 布尔值 | 整数 | 浮点数 | 复数) – 类数组，要评估 PMF 的值

• n (数组 | ndarray | 布尔值 | 数字 | 布尔值 | 整数 | 浮点数 | 复数) – 类数组，分布形状参数

• a (数组 | ndarray | 布尔值 | 数字 | 布尔值 | 整数 | 浮点数 | 复数) – 类数组，分布形状参数

• b (数组 | ndarray | 布尔值 | 数字 | 布尔值 | 整数 | 浮点数 | 复数) – 类数组，分布形状参数

• loc (数组 | ndarray | 布尔值 | 数字 | 布尔值 | 整数 | 浮点数 | 复数) – 类数组，分布偏移参数

返回:

pmf 值数组

返回类型:

数组

另请参阅

jax.scipy.stats.betabinom.logpmf()