jax.scipy.stats.binom.pmf

jax.scipy.stats.binom.pmf(k, n, p, loc=0)

二项分布概率质量函数。

scipy.stats.binom pmf 的 JAX 实现。

二项分布概率质量函数定义为：

\[f(k, n, p) = {n \choose k}p^k(1-p)^{n-k}\]

对于 \(0\le p\le 1\) 和非负整数 \(k\)。

参数:

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

返回:

pmf 值的数组。

返回类型:

Array

另请参阅

jax.scipy.stats.binom.logpmf()