VpfH ddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZdde d e d e d e d e f d Zdde d e d e d e d e f dZdde d e d e d e d e f dZdde d e d e d e d e f dZdde d e d e d e d e f dZdde d e d e d e d e f dZdS))laxN)_const)promote_args_inexact)Array ArrayLike)gammainc gammainccxdflocscalereturnc jtd||||\}}}}t|d}t|d}tjtj|||}tj||}tjtjtj||tj|tj||}tjtjtj |tjtjtj|||} tjtj| tj||} tj tj ||tj | S)aChi-square log probability distribution function. JAX implementation of :obj:`scipy.stats.chi2` ``logpdf``. The chi-square probability distribution function is given by: .. math:: f(x, k) = \begin{cases} \frac{x^{k/2-1}e^{-x/2}}{2^{k/2}\Gamma(k/2)} & x \ge 0 \\ 0 & \mathrm{otherwise} \end{cases} for :math:`k` degrees of freedom, and where :math:`\Gamma` is the :func:`~jax.scipy.special.gamma` function. JAX follows the scipy convention of using ``df`` to denote degrees of freedom. Args: x: arraylike, value at which to evaluate the PDF df: 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.chi2.cdf` - :func:`jax.scipy.stats.chi2.pdf` - :func:`jax.scipy.stats.chi2.sf` - :func:`jax.scipy.stats.chi2.logcdf` - :func:`jax.scipy.stats.chi2.logsf` z chi2.logpdfr )r _lax_constrdivsubmullognegaddlgammajnpwhereltinf) r r r ronetwoy df_on_twokernel nrml_cnst log_probss Y/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/scipy/stats/chi2.pylogpdfr&s6D+=!ReLL!Re1a#1a# gcgaoou%%!gb#) 737379c22CGAJJ??3 P P&gcgcj33CGCGCGCLLRT>CG8Y 7 77cJtjt||||S)aChi-square probability distribution function. JAX implementation of :obj:`scipy.stats.chi2` ``pdf``. The chi-square probability distribution function is given by: .. math:: f(x, k) = \begin{cases} \frac{x^{k/2-1}e^{-x/2}}{2^{k/2}\Gamma(k/2)} & x \ge 0 \\ 0 & \mathrm{otherwise} \end{cases} for :math:`k` degrees of freedom, and where :math:`\Gamma` is the :func:`~jax.scipy.special.gamma` function. JAX follows the scipy convention of using ``df`` to denote degrees of freedom. Args: x: arraylike, value at which to evaluate the PDF df: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of pdf values. See Also: - :func:`jax.scipy.stats.chi2.cdf` - :func:`jax.scipy.stats.chi2.sf` - :func:`jax.scipy.stats.chi2.logcdf` - :func:`jax.scipy.stats.chi2.logpdf` - :func:`jax.scipy.stats.chi2.logsf` )rexpr&r r r rs r%pdfr+Gs#D 2sE** + ++r'c ~td||||\}}}}t|d}ttj||tjt|dtjtj||tj||t|tj S)aQChi-square cumulative distribution function. JAX implementation of :obj:`scipy.stats.chi2` ``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.chi2.pdf`. JAX follows the scipy convention of using ``df`` to denote degrees of freedom. Args: x: arraylike, value at which to evaluate the CDF df: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of cdf values. See Also: - :func:`jax.scipy.stats.chi2.pdf` - :func:`jax.scipy.stats.chi2.sf` - :func:`jax.scipy.stats.chi2.logcdf` - :func:`jax.scipy.stats.chi2.logpdf` - :func:`jax.scipy.stats.chi2.logsf` zchi2.cdfrr) rrrrrclamprrrrr r r rrs r%cdfr/ls>+:q"c5II!Re5!# GBIA g 3 sCG    r'cJtjt||||S)aWChi-square log cumulative distribution function. JAX implementation of :obj:`scipy.stats.chi2` ``logcdf``. 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.chi2.pdf`. JAX follows the scipy convention of using ``df`` to denote degrees of freedom. Args: x: arraylike, value at which to evaluate the CDF df: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of logcdf values See Also: - :func:`jax.scipy.stats.chi2.cdf` - :func:`jax.scipy.stats.chi2.pdf` - :func:`jax.scipy.stats.chi2.sf` - :func:`jax.scipy.stats.chi2.logpdf` - :func:`jax.scipy.stats.chi2.logsf` )rrr/r*s r%logcdfr1s"> QC'' ( ((r'c ~td||||\}}}}t|d}ttj||tjt|dtjtj||tj||t|tj S)a?Chi-square survival function. JAX implementation of :obj:`scipy.stats.chi2` ``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.chi2.cdf`. JAX follows the scipy convention of using ``df`` to denote degrees of freedom. Args: x: arraylike, value at which to evaluate the SF df: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of sf values. See Also: - :func:`jax.scipy.stats.chi2.cdf` - :func:`jax.scipy.stats.chi2.pdf` - :func:`jax.scipy.stats.chi2.logcdf` - :func:`jax.scipy.stats.chi2.logpdf` - :func:`jax.scipy.stats.chi2.logsf` zchi2.sfrr) rrr rrr-rrrrr.s r%sfr3s>+9aS%HH!Re5!# GBIA g 3 sCG    r'cJtjt||||S)aFChi-square log survival function. JAX implementation of :obj:`scipy.stats.chi2` ``logsf``. 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.chi2.cdf`. JAX follows the scipy convention of using ``df`` to denote degrees of freedom. Args: x: arraylike, value at which to evaluate the SF df: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter scale: arraylike, distribution scale parameter Returns: array of logsf values. See Also: - :func:`jax.scipy.stats.chi2.cdf` - :func:`jax.scipy.stats.chi2.pdf` - :func:`jax.scipy.stats.chi2.sf` - :func:`jax.scipy.stats.chi2.logcdf` - :func:`jax.scipy.stats.chi2.logpdf` )rrr3r*s r%logsfr5s"> Ar3&& ' ''r')rr )jaxr jax.numpynumpyrjax._src.lax.laxrrjax._src.numpy.utilrjax._src.typingrrjax.scipy.specialrr r&r+r/r1r3r5r'r%r>s111111444444,,,,,,,,11111111-8-8i-8Y-8Y-89-8UZ-8-8-8-8`",",9",)",)", ",RW",",",",J++9+)+)+ +RW++++\))i)Y)Y)9)UZ))))D++)+++y+QV++++\((Y(I(I()(TY((((((r'