Vpf ddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZmZdde de d e d e fd Zdde de d e d e fd Zdde de d e d e fd ZdS))laxN)_const)promote_args_inexact)Array ArrayLike)xlogygammaln gammaincckmulocreturnc td|||\}}}t|d}tj||}t ||t |dzz |z }t jt jtj ||tj t j ||t j |S)a7Poisson log probability mass function. JAX implementation of :obj:`scipy.stats.poisson` ``logpmf``. The Poisson probability mass function is given by .. math:: f(k) = e^{-\mu}\frac{\mu^k}{k!} and is defined for :math:`k \ge 0` and :math:`\mu \ge 0`. Args: k: arraylike, value at which to evaluate the PMF mu: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of logpmf values. See Also: - :func:`jax.scipy.stats.poisson.cdf` - :func:`jax.scipy.stats.poisson.pmf` poisson.logpmfr) r _lax_constrsubrr jnpwhere logical_orltneroundinf)r r r zerox log_probss \/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/scipy/stats/poisson.pylogpmfrs2$$4aSAA*!R Aq  $ gaoo!ArllWQU^^+b0) 3>#&D//"%&1q"9"9;;=@WHi Q QQcHtjt|||S)a0Poisson probability mass function. JAX implementation of :obj:`scipy.stats.poisson` ``pmf``. The Poisson probability mass function is given by .. math:: f(k) = e^{-\mu}\frac{\mu^k}{k!} and is defined for :math:`k \ge 0` and :math:`\mu \ge 0`. Args: k: arraylike, value at which to evaluate the PMF mu: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of pmf values. See Also: - :func:`jax.scipy.stats.poisson.cdf` - :func:`jax.scipy.stats.poisson.logpmf` )rexpr)r r r s rpmfr#8s 2 2s## $ $$r ctd|||\}}}t|d}tj||}t t jd|z|}t jtj||||S)anPoisson cumulative distribution function. JAX implementation of :obj:`scipy.stats.poisson` ``cdf``. The cumulative distribution function is defined as: .. math:: f_{cdf}(k, p) = \sum_{i=0}^k f_{pmf}(k, p) where :math:`f_{pmf}(k, p)` is the probability mass function :func:`jax.scipy.stats.poisson.pmf`. Args: k: arraylike, value at which to evaluate the CDF mu: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of cdf values. See Also: - :func:`jax.scipy.stats.poisson.pmf` - :func:`jax.scipy.stats.poisson.logpmf` rrr) rrrrr rfloorrr)r r r rrps rcdfr'Tsw4$$4aSAA*!R Aq  $ gaoo! !a%  "%%! 36!T??D! , ,,r )r)jaxr jax.numpynumpyrjax._src.lax.laxrrjax._src.numpy.utilrjax._src.typingrrjax.scipy.specialrr r rr#r'r rr0s8111111444444,,,,,,,,7777777777QQiQYQYQuQQQQB%%9%)%)%E%%%%8--9-)-)-E------r