Vpf4 6ddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZmZmZ dde d e d e d e d e d e f dZ dde d e d e d e d e d e f dZ dde d e d e d e d e d e f dZ dde d e d e d e d e d e f dZ dde d e d e d e d e d e f dZ dde d e d e d e d e d e f dZdS))laxN)_const)promote_args_inexact)Array ArrayLike)betalnbetaincxlogyxlog1pyxablocscalereturnc ,td|||||\}}}}}t|d}t|d}tjt ||}tjtj|||}tjttj|||ttj||tj|} tjtj|| tj |} tj tj tj|tj||tj||tj | } tj tj tj tj||tj||tj||tj| } | S)aiBeta log probability distribution function. JAX implementation of :obj:`scipy.stats.beta` ``logpdf``. The pdf of the beta function is: .. math:: f(x, a, b) = \frac{\Gamma(a + b)}{\Gamma(a)\Gamma(b)} x^{a-1}(1-x)^{b-1} where :math:`\Gamma` is the :func:`~jax.scipy.special.gamma` function, It is defined for :math:`0\le x\le 1` and :math:`b>0`. Args: x: arraylike, value at which to evaluate the PDF a: arraylike, distribution shape parameter b: 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.beta.cdf` - :func:`jax.scipy.stats.beta.pdf` - :func:`jax.scipy.stats.beta.sf` - :func:`jax.scipy.stats.beta.logcdf` - :func:`jax.scipy.stats.beta.logsf` z beta.logpdfr r)r _lax_constrnegrdivsubaddr r logjnpwhere logical_orgtltinflenan) r rrrronezero shape_termylog_linear_term log_probsresultresult_positive_constantss Y/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/scipy/stats/beta.pylogpdfr+s@-]Aq!S%PP!Q31a# Aq  $wva||$$* gcgaoou%%!GE#'!S//155#CGAsOOSWQZZ@@BB/gcgj/::CGENNKK) 9S^CF1cgc5.A.A$B$B$'F1cNN4469gXy J J&!is~cfQPTooWZW]^_aeWfWf7g7g7:veT7J7J)L)LMPWV\^^ ""c Ltjt|||||S)abBeta probability distribution function. JAX implementation of :obj:`scipy.stats.beta` ``pdf``. The pdf of the beta function is: .. math:: f(x, a, b) = \frac{\Gamma(a + b)}{\Gamma(a)\Gamma(b)} x^{a-1}(1-x)^{b-1} where :math:`\Gamma` is the :func:`~jax.scipy.special.gamma` function. It is defined for :math:`0\le x\le 1` and :math:`b>0`. Args: x: arraylike, value at which to evaluate the PDF a: arraylike, distribution shape parameter b: 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.beta.cdf` - :func:`jax.scipy.stats.beta.sf` - :func:`jax.scipy.stats.beta.logcdf` - :func:`jax.scipy.stats.beta.logpdf` - :func:`jax.scipy.stats.beta.logsf` )rexpr+r rrrrs r*pdfr0Fs%@ 1ae,, - --r,c td|||||\}}}}}t||tjt |dtjtj|||t |dS)a>Beta cumulative distribution function JAX implementation of :obj:`scipy.stats.beta` ``cdf``. The cdf is defined as .. math:: f_{cdf}(x, a, b) = \int_{-\infty}^x f_{pdf}(y, a, b)\mathrm{d}y where :math:`f_{pdf}` is the beta distribution probability density function, :func:`jax.scipy.stats.beta.pdf`. Args: x: arraylike, value at which to evaluate the CDF a: arraylike, distribution shape parameter b: 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.beta.pdf` - :func:`jax.scipy.stats.beta.sf` - :func:`jax.scipy.stats.beta.logcdf` - :func:`jax.scipy.stats.beta.logpdf` - :func:`jax.scipy.stats.beta.logsf` zbeta.cdfrr rr rclamprrrr/s r*cdfr4is@-ZAq#uMM!Q3 IA gcgaoou%%A  r,c Ltjt|||||S)aFBeta log cumulative distribution function. JAX implementation of :obj:`scipy.stats.beta` ``logcdf``. The cdf is defined as .. math:: f_{cdf}(x, a, b) = \int_{-\infty}^x f_{pdf}(y, a, b)\mathrm{d}y where :math:`f_{pdf}` is the beta distribution probability density function, :func:`jax.scipy.stats.beta.pdf`. Args: x: arraylike, value at which to evaluate the CDF a: arraylike, distribution shape parameter b: 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.beta.cdf` - :func:`jax.scipy.stats.beta.pdf` - :func:`jax.scipy.stats.beta.sf` - :func:`jax.scipy.stats.beta.logpdf` - :func:`jax.scipy.stats.beta.logsf` )rrr4r/s r*logcdfr6s%@ Q1c5)) * **r,c td|||||\}}}}}t||dtjt |dtjtj|||t |dz S)a1Beta distribution survival function. JAX implementation of :obj:`scipy.stats.beta` ``sf``. The survival function is defined as .. math:: f_{sf}(x, a, b) = 1 - f_{cdf}(x, a, b) where :math:`f_{cdf}(x, a, b)` is the beta cumulative distribution function, :func:`jax.scipy.stats.beta.cdf`. Args: x: arraylike, value at which to evaluate the SF a: arraylike, distribution shape parameter b: 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.beta.cdf` - :func:`jax.scipy.stats.beta.pdf` - :func:`jax.scipy.stats.beta.logcdf` - :func:`jax.scipy.stats.beta.logpdf` - :func:`jax.scipy.stats.beta.logsf` zbeta.sfr rr2r/s r*sfr8s@-Y1aeLL!Q3  A gcgaoou%%A    r,c Ltjt|||||S)a8Beta distribution log survival function. JAX implementation of :obj:`scipy.stats.beta` ``logsf``. The survival function is defined as .. math:: f_{sf}(x, a, b) = 1 - f_{cdf}(x, a, b) where :math:`f_{cdf}(x, a, b)` is the beta cumulative distribution function, :func:`jax.scipy.stats.beta.cdf`. Args: x: arraylike, value at which to evaluate the SF a: arraylike, distribution shape parameter b: 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.beta.cdf` - :func:`jax.scipy.stats.beta.pdf` - :func:`jax.scipy.stats.beta.sf` - :func:`jax.scipy.stats.beta.logcdf` - :func:`jax.scipy.stats.beta.logpdf` )rrr8r/s r*logsfr:s%@ Aq!S%(( ) ))r,)rr )jaxr jax.numpynumpyrjax._src.lax.laxrrjax._src.numpy.utilrjax._src.typingrrjax.scipy.specialrr r r r+r0r4r6r8r:r,r*rCs\111111444444,,,,,,,,============34,#,#i,#I,#),#,#&/,#8=,#,#,#,#`01 . .9 . .y .  .#, .5: . . . .H01))9))y) )#,)5:))))Z34 + +i +I +) + +&/ +8= + + + +H/0)))) )i) )"+)49))))Z23 * *Y *9 * * *%. *7< * * * * * *r,