jax.scipy.stats.bernoulli.logpmf

内容

jax.scipy.stats.bernoulli.logpmf#

jax.scipy.stats.bernoulli.logpmf(k, p, loc=0)[source]#

伯努利对数概率质量函数。

JAX 实现 scipy.stats.bernoulli logpmf

伯努利概率质量函数定义为

\[\begin{split}f(k) = \begin{cases} 1 - p, & k = 0 \\ p, & k = 1 \\ 0, & \mathrm{otherwise} \end{cases}\end{split}\]
参数:
返回值:

logpmf 值数组

返回类型:

Array