§ Vp«fÃã ó†—ddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZd deded ed e fd „Zd deded ed e fd „ZdS)é)ÚlaxN)Ú_const)Úpromote_args_inexact)Úxlog1py)ÚArrayÚ ArrayLikeÚkÚpÚlocÚreturncór—td|||¦«\}}}t|d¦«}t|d¦«}tj||¦«}t tj||¦«| ¦«tj|¦«z}t jtj||¦«t j |¦«S)aòGeometric log probability mass function. JAX implementation of :obj:`scipy.stats.geom` ``logpmf``. The Geometric probability mass function is given by .. math:: f(k) = (1 - p)^{k-1}p for :math:`k\ge 1` and :math:`0 \le p \le 1`. Args: k: arraylike, value at which to evaluate the PMF p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of logpmf values. See Also: :func:`jax.scipy.stats.geom.pmf` z geom.logpmfré) rÚ _lax_constrÚsubrÚlogÚjnpÚwhereÚleÚinf)r r r ÚzeroÚoneÚxÚ log_probss úY/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/scipy/stats/geom.pyÚlogpmfrs”€õ0# =°!°Q¸Ñ<Ô<)€!€QˆÝ AqÑ Ô €$Ý1aÑÔ€#Ý „gˆao„o€!Ý•c”g˜a ‘o”o¨ rÑ*Ô*­S¬W°Q©Z¬ZÑ7€)Ý Œ•3”6˜!˜T‘?”?¥S¤W H¨iÑ 8Ô 8Ð8ócóH—tjt|||¦«¦«S)aëGeometric probability mass function. JAX implementation of :obj:`scipy.stats.geom` ``pmf``. The Geometric probability mass function is given by .. math:: f(k) = (1 - p)^{k-1}p for :math:`k\ge 1` and :math:`0 \le p \le 1`. Args: k: arraylike, value at which to evaluate the PMF p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of pmf values. See Also: :func:`jax.scipy.stats.geom.logpmf` )rÚexpr)r r r s rÚpmfr7s €õ0 Œ•˜˜1˜cÑ"Ô"Ñ #Ô #Ð#r)r)ÚjaxrÚ jax.numpyÚnumpyrÚjax._src.lax.laxrrÚjax._src.numpy.utilrÚjax.scipy.specialrÚjax._src.typingrrrr©rrúr(sîððÐÐÐÐÐØÐÐÐÐÐØ1Ð1Ð1Ð1Ð1Ð1Ø4Ð4Ð4Ð4Ð4Ð4Ø%Ð%Ð%Ð%Ð%Ð%Ø,Ð,Ð,Ð,Ð,Ð,Ð,Ð,ð9ð9ˆið9˜Ið9¨Ið9¸eð9ð9ð9ð9ð@$ð$ˆ9ð$˜ð$¨ð$¸5ð$ð$ð$ð$ð$ð$r