jax.scipy.stats.bernoulli.logpmf#

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

Bernoulli 對數機率質量函數。

scipy.stats.bernoulli logpmf 的 JAX 實作

Bernoulli 機率質量函數定義如下：

\[\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) – 類陣列 (arraylike)，用於評估 PMF 的值

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

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

返回：

logpmf 值陣列

返回類型：

Array

參見