jax.scipy.stats.binom.logpmf

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

二项分布的对数概率质量函数。

scipy.stats.binom logpmf 的 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 ) – 类数组，分布偏移参数。

返回:

对数概率质量函数 (logpmf) 值的数组。

返回类型:

Array

另请参阅

jax.scipy.stats.binom.pmf()