jax.scipy.stats.bernoulli.pmf#
- jax.scipy.stats.bernoulli.pmf(k, p, loc=0)[source]#
Probability mass function at k of the given RV.
LAX-backend implementation of
scipy.stats._distn_infrastructure.pmf()
.Original docstring below.
- Parameters:
k (array_like) – Quantiles.
arg1 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
arg2 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
arg3 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
... (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)
loc (array_like, optional) – Location parameter (default=0).
p (Array | ndarray | bool_ | number | bool | int | float | complex)
- Returns:
pmf – Probability mass function evaluated at k
- Return type:
array_like