Vpf ddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z d de de de d e d e f d Zd de de de d e d e f d ZdS))laxN)promote_args_inexact)gammalnxlogyxlog1py)Array ArrayLikeknplocreturnc td||||\}}}}tj||}tjt|dztjt|dzt||z dz}tjt ||t tj||tj|}tj||}tj tj ||tj |||zdzz|tj S)aEBinomial log probability mass function. JAX implementation of :obj:`scipy.stats.binom` ``logpmf``. The binomial probability mass function is defined as .. math:: f(k, n, p) = {n \choose k}p^k(1-p)^{n-k} for :math:`0\le p\le 1` and non-negative integers :math:`k`. Args: k: arraylike, value at which to evaluate the PMF n: arraylike, distribution shape parameter p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of logpmf values. See Also: :func:`jax.scipy.stats.binom.pmf` z binom.logpmf) rrsubraddrrnegjnpwheregeltinf)r r r r y comb_termlog_linear_term log_probss Z/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/scipy/stats/binom.pylogpmfrs2&naAsCC,!Q3 gaoo!g a!enn gga!ennga!eai0011)GE!QKKA )K)KLL/gi11) 36!S>>CF1cAgk$:$::Ix P PPcJtjt||||S)a@Binomial probability mass function. JAX implementation of :obj:`scipy.stats.binom` ``pmf``. The binomial probability mass function is defined as .. math:: f(k, n, p) = {n \choose k}p^k(1-p)^{n-k} for :math:`0\le p\le 1` and non-negative integers :math:`k`. Args: k: arraylike, value at which to evaluate the PMF n: arraylike, distribution shape parameter p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of pmf values. See Also: :func:`jax.scipy.stats.binom.logpmf` )rexpr)r r r r s rpmfr":s"2 1a%% & &&r)r)jaxr jax.numpynumpyrjax._src.numpy.utilrjax._src.scipy.specialrrrjax._src.typingrr rr"rrr*s444444::::::::::,,,,,,,,!Q!Qi!QI!Q)!Q)!QE!Q!Q!Q!QH''9''y'y'''''''r