jax.scipy.stats.bernoulli.pmf

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

伯努利概率质量函数。

JAX 对 scipy.stats.bernoulli 的 pmf 的实现。

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

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

参数:

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

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

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

返回:

PMF值的数组

返回类型:

Array

另请参阅

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

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

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