uki ddlmZddlmZddlmZddlmZm Z d de de de defd Z d de de de defd Z d de de de defd Z y ))lax)_const)promote_args_inexact)Array ArrayLikexlocscalereturnc Xtd|||\}}}t|d}tjtjtj |||}tj tj|tjtj||S)a Laplace log probability distribution function. JAX implementation of :obj:`scipy.stats.laplace` ``logpdf``. The Laplace probability distribution function is given by .. math:: f(x) = \frac{1}{2} e^{-|x|} Args: x: arraylike, value at which to evaluate the PDF loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of logpdf values. See Also: - :func:`jax.scipy.stats.laplace.cdf` - :func:`jax.scipy.stats.laplace.pdf` zlaplace.logpdf) r _lax_constrdivabssubnegaddlogmul)rr r two linear_terms W/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/scipy/stats/laplace.pylogpdfrsx.''7CG-!S%1a#30%8+ cggcggc5.A&BC DDcBtjt|||S)aLaplace probability distribution function. JAX implementation of :obj:`scipy.stats.laplace` ``pdf``. The Laplace probability distribution function is given by .. math:: f(x) = \frac{1}{2} e^{-|x|} Args: x: arraylike, value at which to evaluate the PDF loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of pdf values. See Also: - :func:`jax.scipy.stats.laplace.cdf` - :func:`jax.scipy.stats.laplace.logpdf` )rexpr)rr r s rpdfr2s. 3& ''rctd|||\}}}t|d}t|d}t|d}tjtj|||}tj tj ||tj|tj|tj|tj|tjtj|S)a`Laplace cumulative distribution function. JAX implementation of :obj:`scipy.stats.laplace` ``cdf``. The cdf is defined as .. math:: f_{cdf}(x, k) = \int_{-\infty}^x f_{pdf}(y, k)\mathrm{d}y where :math:`f_{pdf}` is the probability density function, :func:`jax.scipy.stats.laplace.pdf`. Args: x: arraylike, value at which to evaluate the CDF loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of cdf values. See Also: - :func:`jax.scipy.stats.laplace.pdf` - :func:`jax.scipy.stats.laplace.logpdf` z laplace.cdfg?r) rrrrrselectlerrr)rr r halfonezerodiffs rcdfr&Ls4'}aeD-!S% As $1a# Aq $ C% ($ CFF4&GGD#''$-0GGCswwswwt}/E!FG IIrN)rr) jax._srcrjax._src.lax.laxrrjax._src.numpy.utilrjax._src.typingrrrrr&rrr,s14,EiEiEIEeE:(9(9((5(4!I9!I9!I!I5!Ir