uki ddlmZddlmZddlmZddlmZddl m Z m Z ddl m Z mZddeded ed e fd Zddeded ed e fd Zddeded ed e fd Zddeded ed e fdZddeded ed e fdZddeded ed e fdZy))lax)numpy)_const)promote_args_inexact)expitlogit)Array ArrayLikexlocscalereturnc td|||\}}}tjtj|||}t |d}tj||}tjtj tj |tj|tj |tj|S)aLogistic log probability distribution function. JAX implementation of :obj:`scipy.stats.logistic` ``logpdf``. The logistic probability distribution function is given by .. math:: f(x) = \frac{e^{-x}}{(1 + e^{-x})^2} Args: x: arraylike, value at which to evaluate the PDF a: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of logpdf values. See Also: - :func:`jax.scipy.stats.logistic.cdf` - :func:`jax.scipy.stats.logistic.pdf` - :func:`jax.scipy.stats.logistic.sf` - :func:`jax.scipy.stats.logistic.isf` - :func:`jax.scipy.stats.logistic.ppf` zlogistic.logpdf) rrdivsub _lax_constmulnegjnp logaddexplog)r r r twohalf_xs X/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/scipy/stats/logistic.pylogpdfrs6''8!S%H-!S% ggcggaou%!1a# 771c?& s}}VSWWV_'MNPSPWPWX]P^ __cBtjt|||S)aLogistic probability distribution function. JAX implementation of :obj:`scipy.stats.logistic` ``pdf``. The logistic probability distribution function is given by .. math:: f(x) = \frac{e^{-x}}{(1 + e^{-x})^2} 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.logistic.cdf` - :func:`jax.scipy.stats.logistic.sf` - :func:`jax.scipy.stats.logistic.isf` - :func:`jax.scipy.stats.logistic.logpdf` - :func:`jax.scipy.stats.logistic.ppf` )rexprr r r s rpdfr!:s4 3& ''rctd|||\}}}tjtjt |||S)aLogistic distribution percent point function. JAX implementation of :obj:`scipy.stats.logistic` ``ppf``. The percent point function is defined as the inverse of the cumulative distribution function, :func:`jax.scipy.stats.logistic.cdf`. Args: x: arraylike, value at which to evaluate the PPF loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of ppf values. See Also: - :func:`jax.scipy.stats.logistic.cdf` - :func:`jax.scipy.stats.logistic.pdf` - :func:`jax.scipy.stats.logistic.sf` - :func:`jax.scipy.stats.logistic.isf` - :func:`jax.scipy.stats.logistic.logpdf` z logistic.ppf)rraddrrr s rppfr$Ws:.'~q#uE-!S% q5)3 //rc td|||\}}}ttjtjtj |||S)aLogistic distribution survival function. JAX implementation of :obj:`scipy.stats.logistic` ``sf`` The survival function is defined as .. math:: f_{sf}(x, k) = 1 - f_{cdf}(x, k) where :math:`f_{cdf}(x, k)` is the cumulative distribution function, :func:`jax.scipy.stats.logistic.cdf`. Args: x: arraylike, value at which to evaluate the SF loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of sf values. See Also: - :func:`jax.scipy.stats.logistic.cdf` - :func:`jax.scipy.stats.logistic.pdf` - :func:`jax.scipy.stats.logistic.isf` - :func:`jax.scipy.stats.logistic.logpdf` - :func:`jax.scipy.stats.logistic.ppf` z logistic.sf)rrrrrrr s rsfr&rsC:'}aeD-!S% swwswwswwq#67 88rc td|||\}}}tjtjtjt |||S)awLogistic distribution inverse survival function. JAX implementation of :obj:`scipy.stats.logistic` ``isf``. Returns the inverse of the survival function, :func:`jax.scipy.stats.logistic.sf`. Args: x: arraylike, value at which to evaluate the ISF loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of isf values. See Also: - :func:`jax.scipy.stats.logistic.cdf` - :func:`jax.scipy.stats.logistic.pdf` - :func:`jax.scipy.stats.logistic.sf` - :func:`jax.scipy.stats.logistic.logpdf` - :func:`jax.scipy.stats.logistic.ppf` z logistic.isf)rrr#rrrr s risfr(sC.'~q#uE-!S% q*E2C 88rctd|||\}}}ttjtj|||S)aLogistic cumulative distribution function. JAX implementation of :obj:`scipy.stats.logistic` ``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.logistic.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.logistic.pdf` - :func:`jax.scipy.stats.logistic.sf` - :func:`jax.scipy.stats.logistic.isf` - :func:`jax.scipy.stats.logistic.logpdf` - :func:`jax.scipy.stats.logistic.ppf` z logistic.cdf)rrrrrr s rcdfr*s::'~q#uE-!S% swwswwq#. //rN)r)jax._srcrrrjax._src.lax.laxrrjax._src.numpy.utilrjax._src.scipy.specialrrjax._src.typingr r rr!r$r&r(r*rrr2s !14/,`i`i`I`e`D(9(9((5(:0909005069)9)9 9%9B99999959609090050r