jax.scipy.stats.bernoulli.logpmf

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

伯努利分布的对数概率质量函数。

JAX实现的 scipy.stats.bernoulli logpmf

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

\[\begin{split}f(k) = \begin{cases} 1 - p, & k = 0 \\ p, & k = 1 \\ 0, & \mathrm{其他情况} \end{cases}\end{split}\]

参数:

• k ( Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，用于评估 PMF 的值

• p ( Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布形状参数

• loc ( Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布偏移量

返回:

对数概率质量函数值的数组

返回类型:

数组

另请参阅

• jax.scipy.stats.bernoulli.cdf()

• jax.scipy.stats.bernoulli.pmf()

• jax.scipy.stats.bernoulli.ppf()