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) – 類陣列，評估 PMF 的值

• p (Array | ndarray | bool | number | bool | int | float | complex) – 類陣列，分布形狀參數

• loc (Array | ndarray | bool | number | bool | int | float | complex) – 類陣列，分布偏移

返回:

pmf 值陣列

返回類型:

Array

參見

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

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

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