jax.scipy.stats.bernoulli.pmf
Contents
jax.scipy.stats.bernoulli.pmf#
- jax.scipy.stats.bernoulli.pmf(k,p,loc=0)[source]#
Bernoulli probability mass function.
JAX implementation of
scipy.stats.bernoullipmfThe Bernoulli probability mass function is defined as
\[\begin{split}f(k) = \begin{cases} 1 - p, & k = 0 \\ p, & k = 1 \\ 0, & \mathrm{otherwise}\end{cases}\end{split}\]- Parameters:
k (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray) – arraylike, value at which to evaluate the PMF
p (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray) – arraylike, distribution shape parameter
loc (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray) – arraylike, distribution offset
- Returns:
array of pmf values
- Return type:
Contents