o Õ0 i¹¹ã@søddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl Z ddl m Z ddl mZddlmZddlmZddl mZddlZdZd ZGd d „d eƒZdd d „Ziaejdd„ƒZdd„Zdd„Zdd„ZdS)éN)Ú LinAlgError)Ú_core)Úcuda)Údevice)Ú_kernels)Ú_utillÿÿlÿÿÿÿc@s@eZdZdZdŽdd„Zdd„Zdd„Zdefd d „Zde fd d „Z defd d„Z defdd„Z ddefdd„Z defdd„Zddefdd„Zde fdd„Zde fdd„Ze dddd¡Zd ddefd!d"„Zd ddefd#d$„Zd ddefd%d&„Zde fd'd(„Zdd)d*d+efd,d-„Zde fd.d/„Zd ddefd0d1„Zdefd2d3„Zdefd4d5„Zdefd6d7„Zdde fd8d9„Zdefd:d;„Z dd?„Z"e d@dAdBdC¡Z#dDdE„Z$defdFdG„Z%ddefdHdI„Z&e dJdKdLdM¡Z'e dJdNdOdP¡Z(dQdR„Z)dSdT„Z*dUdV„Z+e dWdXdYdZ¡Z,dd[d\„Z-defd]d^„Z.defd_d`„Z/defdadb„Z0defdcdd„Z1defdedf„Z2ddgdh„Z3e didAdjdk¡Z4defdldm„Z5e dndAdodp¡Z6d ddefdqdr„Z7defdsdt„Z8e dudvdwdx¡Z9defdydz„Z:defd{d|„Z;de fd}d~„Zd„d…„Z?d†d‡„Z@e dddˆd‰¡ZAd ddefdŠd‹„ZBdde fdŒd„ZCdS)‘Ú RandomStateaVPortable container of a pseudo-random number generator. An instance of this class holds the state of a random number generator. The state is available only on the device which has been current at the initialization of the instance. Functions of :mod:`cupy.random` use global instances of this class. Different instances are used for different devices. The global state for the current device can be obtained by the :func:`cupy.random.get_random_state` function. Args: seed (None or int): Seed of the random number generator. See the :meth:`~cupy.random.RandomState.seed` method for detail. method (int): Method of the random number generator. Following values are available:: cupy.cuda.curand.CURAND_RNG_PSEUDO_DEFAULT cupy.cuda.curand.CURAND_RNG_PSEUDO_XORWOW cupy.cuda.curand.CURAND_RNG_PSEUDO_MRG32K3A cupy.cuda.curand.CURAND_RNG_PSEUDO_MTGP32 cupy.cuda.curand.CURAND_RNG_PSEUDO_MT19937 cupy.cuda.curand.CURAND_RNG_PSEUDO_PHILOX4_32_10 NcCsNddlm}|dur |j}| |¡|_t ||j|j¡|_||_ |  |¡dS)Nr©Úcurand) Úcupy_backends.cuda.libsr ZCURAND_RNG_PSEUDO_DEFAULTZcreateGeneratorÚ _generatorÚweakrefÚfinalizeZdestroyGeneratorÚ _finalizerÚmethodÚseed)Úselfrrr ©rúb/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupy/random/_generator.pyÚ__init__8s   ÿzRandomState.__init__cCs|j|t|_dS©N)Ú_rk_seedÚ _UINT64_MAX)rÚsizerrrÚ _update_seedCszRandomState._update_seedcGs’t |¡}tj |¡}|ddkr(tj||d}||j|jj|j g|¢RŽ|Stj|df|d}||j|jj|j g|¢RŽ|d|…  |¡S)Nér©Údtypeé) rÚget_sizeZinternalÚprodÚcupyÚemptyr ÚdataÚptrrÚreshape)rÚfuncrrÚargsZ element_sizeÚoutrrrÚ_generate_normalFs   zRandomState._generate_normalcCó\t |¡t |¡}}|durt ||¡j}tj||d}t |||j|¡| |j ¡|S)zÎReturns an array of samples drawn from the beta distribution. .. seealso:: - :func:`cupy.random.beta` for full documentation - :meth:`numpy.random.RandomState.beta` N©Úshaper) r!ÚasarrayÚ broadcastr,r"rZ beta_kernelrrr)rÚaÚbrrÚyrrrÚbetaYó zRandomState.betacCr*)zÚReturns an array of samples drawn from the binomial distribution. .. seealso:: - :func:`cupy.random.binomial` for full documentation - :meth:`numpy.random.RandomState.binomial` Nr+) r!r-r.r,r"rZbinomial_kernelrrr©rÚnÚprrr1rrrÚbinomialhr3zRandomState.binomialcCóFt |¡}|dur |j}tj||d}t ||j|¡| |j¡|S)zÞReturns an array of samples drawn from the chi-square distribution. .. seealso:: - :func:`cupy.random.chisquare` for full documentation - :meth:`numpy.random.RandomState.chisquare` Nr+) r!r-r,r"rZchisquare_kernelrrr©rÚdfrrr1rrrÚ chisquarewó  zRandomState.chisquarecCs‚t |¡}|dur |j}nt|ttjfƒr|f|j}n||j7}tj||d}t ||j |¡||j ddd}|  |j ¡|S)zÝReturns an array of samples drawn from the dirichlet distribution. .. seealso:: - :func:`cupy.random.dirichlet` for full documentation - :meth:`numpy.random.RandomState.dirichlet` Nr+éÿÿÿÿT)ÚaxisZkeepdims) r!r-r,Ú isinstanceÚintÚintegerr"rÚstandard_gamma_kernelrÚsumrr)rÚalpharrr1rrrÚ dirichlet†s   zRandomState.dirichletçð?cCsFt ||¡}|dk ¡rtdƒ‚|dur|j}| ||¡}||9}|S)a.Returns an array of samples drawn from a exponential distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.exponential` for full documentation - :meth:`numpy.random.RandomState.exponential` rú scale < 0N)r!r-ÚanyÚ ValueErrorr,Ústandard_exponential©rÚscalerrÚxrrrÚ exponentialšs  zRandomState.exponentialcCr*)zÅReturns an array of samples drawn from the f distribution. .. seealso:: - :func:`cupy.random.f` for full documentation - :meth:`numpy.random.RandomState.f` Nr+) r!r-r.r,r"rZf_kernelrrr)rÚdfnumÚdfdenrrr1rrrÚf®r3z RandomState.fcCsbt |¡t |¡}}|durt ||¡j}tj||d}t ||j|¡||9}| |j ¡|S)zÏReturns an array of samples drawn from a gamma distribution. .. seealso:: - :func:`cupy.random.gamma` for full documentation - :meth:`numpy.random.RandomState.gamma` Nr+) r!r-r.r,r"rrBrrr)rr,rLrrr1rrrÚgamma½s zRandomState.gammacCr8)zÝReturns an array of samples drawn from the geometric distribution. .. seealso:: - :func:`cupy.random.geometric` for full documentation - :meth:`numpy.random.RandomState.geometric` Nr+) r!r-r,r"rZgeometric_kernelrrr©rr6rrr1rrrÚ geometricÍr<zRandomState.geometriccCslt |¡t |¡t |¡}}}|durt |||¡j}tj||d}t ||||j|¡| |j ¡|S)zìReturns an array of samples drawn from the hypergeometric distribution. .. seealso:: - :func:`cupy.random.hypergeometric` for full documentation - :meth:`numpy.random.RandomState.hypergeometric` Nr+) r!r-r.r,r"rZhypergeometric_kernelrrr)rZngoodZnbadZnsamplerrr1rrrÚhypergeometricÜsÿ zRandomState.hypergeometriczT x, T loc, T scalezT yz>y = loc + scale * ((x <= 0.5) ? log(x + x): -log(x + x - 1.0))Zcupy_laplace_kernelçcCsNt ||¡}t ||¡}|durt ||¡j}| ||¡}t ||||¡|S)z×Returns an array of samples drawn from the laplace distribution. .. seealso:: - :func:`cupy.random.laplace` for full documentation - :meth:`numpy.random.RandomState.laplace` N)r!r-r.r,Ú_random_sample_rawrÚ_laplace_kernel©rÚlocrLrrrMrrrÚlaplaceñs   zRandomState.laplacecCs’t |¡t |¡}}|durt ||¡j}tj||d}t |j|¡| |j ¡d||}tj ||dtj |||dtj |||d|S)zÚReturns an array of samples drawn from the logistic distribution. .. seealso:: - :func:`cupy.random.logistic` for full documentation - :meth:`numpy.random.RandomState.logistic` Nr+rF©r() r!r-r.r,r"rZopen_uniform_kernelrrrÚlogÚmultiplyÚaddrYrrrÚlogistics  zRandomState.logisticcCs‚ddlm}tdd„||fDƒƒr"| ||||¡}tj||d|S|dur(d}t|ƒ}|jdkr5|j}n|j }|  |||||¡S) zÝReturns an array of samples drawn from a log normal distribution. .. seealso:: - :func:`cupy.random.lognormal` for full documentation - :meth:`numpy.random.RandomState.lognormal` rr css|] }t|tjƒVqdSr)r?r!Úndarray)Ú.0ÚargrrrÚ s€z(RandomState.lognormal..r\NrrQ) r r rHÚnormalr!ÚexpÚ_check_and_get_dtypeÚcharZgenerateLogNormalZgenerateLogNormalDoubler))rÚmeanÚsigmarrr rMr&rrrÚ lognormals  zRandomState.lognormalcCsrt |¡}t |dk¡rtdƒ‚t |dk¡rtdƒ‚|dur"|j}tj||d}t ||j|¡|  |j ¡|S)a*Returns an array of samples drawn from a log series distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.logseries` for full documentation - :meth:`numpy.random.RandomState.logseries` rzp <= 0.0rzp >= 1.0Nr+) r!r-rHrIr,r"rZlogseries_kernelrrrrSrrrÚ logseries*s  zRandomState.logseriesÚignoreg:Œ0âŽyE>Úcholeskyc Cs>t d¡tj||d}tj||d}|durg}nt|ttjfƒr&|g}n|}t|jƒdkr3t dƒ‚t|jƒdksD|jd|jdkrHt dƒ‚|jd|jdkrVt d ƒ‚t |dd…ƒ} |   |jd¡|d vrnt d ƒ‚|d kr~|d kr~|dkr~t dƒ‚|d krt j dd‡ztj |¡} Wnvtyt j d d]|dkrê|dkrºtj |¡\} } t | | k¡ } |dkr×tj |¡\} } }tjt |j| |¡|||d} | t t | ¡¡} | sét dt¡n t dt¡tj |¡} Wdƒn 1swYYnwWdƒn 1swYnbt j |dSz>|dkr.tj |¡} n0|dkrFtj |¡\} } | t t | ¡¡} n|dkr^tj |¡\} } }| t t | ¡¡} Wn tyktdƒ‚wWdƒn 1swwY|j| |d d|jd¡}t | |j¡}|j}||7}t| ƒ|_|S)aeReturns an array of samples drawn from the multivariate normal distribution. .. warning:: This function calls one or more cuSOLVER routine(s) which may yield invalid results if input conditions are not met. To detect these invalid results, you can set the `linalg` configuration to a value that is not `ignore` in :func:`cupyx.errstate` or :func:`cupyx.seterr`. .. seealso:: - :func:`cupy.random.multivariate_normal` for full documentation - :meth:`numpy.random.RandomState.multivariate_normal` z+cupy.random.RandomState.multivariate_normalrNrzmean must be 1 dimensionalrrz$cov must be 2 dimensional and squarez"mean and cov must have same length>ÚeighrnÚsvdz1method must be one of {'eigh', 'svd', 'cholesky'}rmÚwarnÚraisez3check_valid must equal 'warn', 'raise', or 'ignore')Úlinalgrnrorp)ZrtolZatolz?covariance is not positive-semidefinite, output may be invalid.z=covariance is not positive-semidefinite, output *is* invalid.z_Matrix is not positive definite; if matrix is positive-semidefinite, set'check_valid' to 'warn'r=)rZ experimentalr!r-r?r@rAÚlenr,rIÚlistÚappendÚcupyxZerrstatersrnrrorHrpÚallcloseÚdotÚTÚsqrtÚabsÚwarningsrqÚRuntimeWarningÚstandard_normalr%Útuple)rriZcovrZ check_validZtolrrr,Z final_shapeÚdecompÚsÚuZpsdZvhrMrrrÚmultivariate_normalBs¦ "ÿÿ ÿþ€þ €í€ÿ€ý€   €ÿ€õÿÿ zRandomState.multivariate_normalcCszt |¡}t |¡}t |dk¡rtdƒ‚t |dk¡r tdƒ‚t |dk¡r+tdƒ‚| |d|||¡}|j||dS)aBReturns an array of samples drawn from the negative binomial distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.negative_binomial` for full documentation - :meth:`numpy.random.RandomState.negative_binomial` rzn <= 0zp < 0rzp > 1r)r!r-rHrIrRÚpoissonr4rrrÚnegative_binomial¢s zRandomState.negative_binomialcCsÌddlm}t|ƒ}|durt ||¡j}|jdkr|j}n|j}t |tj ƒrB|  |||dd¡}tj |||dtj |||d|St |tj ƒr[|  |||d|¡}tj |||d|S|  |||||¡}|S)zÇReturns an array of normally distributed samples. .. seealso:: - :func:`cupy.random.normal` for full documentation - :meth:`numpy.random.RandomState.normal` rr NrQrVrFr\)r r rgr!r.r,rhZgenerateNormalZgenerateNormalDoubler?rar)r^r_)rrZrLrrr r&rMrrrre¸s$    ûÿzRandomState.normalcCsNt |¡}|dur |j}| ||¡}tj||dtj| ||d|dS)z×Returns an array of samples drawn from the pareto II distribution. .. seealso:: - :func:`cupy.random.pareto` for full documentation - :meth:`numpy.random.RandomState.pareto` Nr\r)r!r-r,rWr]rf©rr/rrrMrrrÚparetoÔs  zRandomState.paretocCsˆt |¡t |¡}}t |dk¡rtdƒ‚t |dk¡r!tdƒ‚|dur,t ||¡j}tj||d}t |||j |¡|  |j ¡|S)aTReturns an array of samples drawn from the noncentral chi-square distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.noncentral_chisquare` for full documentation - :meth:`numpy.random.RandomState.noncentral_chisquare` rzdf <= 0únonc < 0Nr+) r!r-rHrIr.r,r"rZnoncentral_chisquare_kernelrrr)rr:Únoncrrr1rrrÚnoncentral_chisquareãs  z RandomState.noncentral_chisquarecCs®t |¡t |¡t |¡}}}t |dk¡rtdƒ‚t |dk¡r'tdƒ‚t |dk¡r2tdƒ‚|dur>t |||¡j}tj||d}t ||||j |¡|  |j ¡|S)a3Returns an array of samples drawn from the noncentral F distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.noncentral_f` for full documentation - :meth:`numpy.random.RandomState.noncentral_f` rz dfnum <= 0z dfden <= 0r‰Nr+) r!r-rHrIr.r,r"rZnoncentral_f_kernelrrr)rrOrPrŠrrr1rrrÚ noncentral_fûs ÿ zRandomState.noncentral_fcCr8)z×Returns an array of samples drawn from the poisson distribution. .. seealso:: - :func:`cupy.random.poisson` for full documentation - :meth:`numpy.random.RandomState.poisson` Nr+) r!r-r,r"rZpoisson_kernelrrr)rZlamrrr1rrrr…r<zRandomState.poissoncCsvt |¡}t |dk¡rtdƒ‚|dur|j}|j||d}tj| |dtjd| |dtj|d||d|S)aReturns an array of samples drawn from the power distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.power` for full documentation - :meth:`numpy.random.RandomState.power` rúa < 0N©rrr\rrF) r!r-rHrIr,rJrfr_Úpowerr‡rrrr$s zRandomState.powercOó4| dt¡}|rtdd | ¡¡ƒ‚|j||dS)zÍReturns uniform random values over the interval ``[0, 1)``. .. seealso:: - :func:`cupy.random.rand` for full documentation - :meth:`numpy.random.RandomState.rand` rz*rand() got unexpected keyword arguments %sú, rŽ)ÚpopÚfloatÚ TypeErrorÚjoinÚkeysÚ random_sample©rrÚkwargrrrrÚrand:ó  ÿzRandomState.randcOr)zÆReturns an array of standard normal random values. .. seealso:: - :func:`cupy.random.randn` for full documentation - :meth:`numpy.random.RandomState.randn` rz+randn() got unexpected keyword arguments %sr‘rŽ)r’r“r”r•r–rer˜rrrÚrandnHr›zRandomState.randnÚzT xzx = (x == (T)1) ? 0 : xZcupy_random_x_mod_1cCsRddlm}t|ƒ}tj||d}|jdkr|j}n|j}||j|j j |j ƒ|S)Nrr rrQ) r r rgr!r"rhZgenerateUniformZgenerateUniformDoubler r#r$r)rrrr r(r&rrrrWYs  zRandomState._random_sample_rawcCs&|durd}| ||¡}t |¡|S)zãReturns an array of random values over the interval ``[0, 1)``. .. seealso:: - :func:`cupy.random.random_sample` for full documentation - :meth:`numpy.random.RandomState.random_sample` Nr)rWrÚ _mod1_kernel)rrrr(rrrr—es   zRandomState.random_samplecCszt |¡}|dur |j}t |dk¡rtdƒ‚| ||¡}tj||d}tj|d|d}tj||d}tj|||d}|S)a%Returns an array of samples drawn from a rayleigh distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.rayleigh` for full documentation - :meth:`numpy.random.RandomState.rayleigh` NrrGr\gÀ) r!r-r,rHrIrWr]r^r{rKrrrÚrayleighss  zRandomState.rayleighz T max, T mxzT outz3out = max - (mx != max ? (max - mx) % (mx + 1) : 0)Z cupy_random_interval_upper_limitzT samplez$if (mx != max) { sample %= mx + 1; }Z"cupy_random_interval_sample_modulocCs¾|durd}nt|tƒr|f}t |¡}|rX|dkr"tj|tjdS|dkr-td |¡ƒ‚|t kr?tj}|t d|dƒ}nY|t krQtj }|t d|dƒ}nGtd  |¡ƒ‚|j }|tj kse|tjkrvtj}|j|d d }| t |¡}n"|tjks€|tj kr‘tj }|j|d d }| t |¡}ntd  |¡ƒ‚t tj|d¡}|dkr«tj||dS| ||¡}|rÒ|d}|d| ¡d>krÒd| ¡>d} || M}| |¡S| ||d ¡} | j} | dkrét |¡sé|| }| dkr=|r÷t| d dƒn|j} | | |¡} | | |d¡}|j}|| kr| |d| …|| <d} n| ||| d|…<| |d…} |s5||d…}| |8} | dksî|rE||;}n|tjkrS| t ||¡}n| t ||¡}| |¡S)a`Generate multiple integers independently sampled uniformly from ``[0, mx]``. Args: mx (int): Upper bound of the interval size (None or int or tuple): Shape of the array or the scalar returned. Returns: int or cupy.ndarray: If ``None``, an :class:`cupy.ndarray` with shape ``()`` is returned. If ``int``, 1-D array of length size is returned. If ``tuple``, multi-dimensional array with shape ``size`` is returned. Currently, only 32 bit or 64 bit integers can be sampled. Nrrrz$mx must be non-negative (actual: {})ìrìz+mx must be within uint64 range (actual: {})F©Úcopyzdtype must be integer, got: {}riT)r?r@ÚnumpyÚisscalarr!ZzerosÚuint32rIÚformatÚ _UINT32_MAXrÚuint64rÚint32ÚastypeÚ_interval_upper_limitÚint64Ú functoolsÚreduceÚoperatorÚmulr"Ú_curand_generateÚ bit_lengthr%Ú _get_indicesrÚmaxÚ_interval_sample_modulo)rZmxrZ is_mx_scalarrÚ upper_limitZn_sampleÚsampleZmx1ÚmaskZ ng_indicesZn_ngZ n_supplementZ supplementZ ok_indicesZn_okrrrÚ _interval”sŒ  ÿÿÿ   ÿÿ    î   zRandomState._intervalcCsDddlm}tj|f|d}|jtjdj}| |j |j j |¡|S)Nrr r) r r r!r"Úviewr¤r¦rÚgenerater r#r$)rÚnumrr r¸Zsize32rrrr²÷s zRandomState._curand_generatecCsh|jdkrtjntj}|r||kn||k}tj||d}~tjt|dƒf|d}|j|||jd|S)Nr rr=©r) rr¤r¦r©r!Zcumsumr"r@Ú_kernel_get_indices)rr¸r·ZcondrÚflagsZcsumÚindicesrrrr´szRandomState._get_indicesz raw U csumz raw U indiceszo int j = 0; if (i > 0) { j = csum[i-1]; } if (csum[i] > j) { indices[j] = i; } Zcupy_get_indicescCsúddlm}|dur+zt t d¡¡}t|dƒ}WnGty*t ¡dt }Yn7wt |t j ƒrBtt j|dd ¡dd…dƒ}nt  |¡}|jjd vrQtd ƒ‚t|ƒ}|dks]|d kratd ƒ‚| |j|¡|j|j|jfvrx| |jd¡||_dS) zÎResets the state of the random number generator with a seed. .. seealso:: - :func:`cupy.random.seed` for full documentation - :meth:`numpy.random.RandomState.seed` rr Nééi@BF)ÚusedforsecurityZbiuzSeed must be an integer.r¡z/Seed must be an integer between 0 and 2**64 - 1)r r ÚbinasciiÚhexlifyÚosÚurandomr@ÚNotImplementedErrorÚtimerr?r¤raÚhashlibÚmd5Ú hexdigestr-rÚkindr”rIZsetPseudoRandomGeneratorSeedr rZCURAND_RNG_PSEUDO_MT19937ZCURAND_RNG_PSEUDO_MTGP32ZsetGeneratorOffsetr)rrr Zseed_strZseed_arrrrrrs<  ÿ ÿ ÿÿ  ÿÿ zRandomState.seedcCs"|j||d}t tj|d¡S)zïReturns an array of samples drawn from the standard cauchy distribution. .. seealso:: - :func:`cupy.random.standard_cauchy` for full documentation - :meth:`numpy.random.RandomState.standard_cauchy` rŽgà?)Úuniformr!ÚtanÚpi©rrrrMrrrÚstandard_cauchy7szRandomState.standard_cauchycCs(|durd}| ||¡}tj||d S)z÷Returns an array of samples drawn from the standard exp distribution. .. seealso:: - :func:`cupy.random.standard_exponential` for full documentation - :meth:`numpy.random.RandomState.standard_exponential` Nrr\)rWr!r]rÒrrrrJAs z RandomState.standard_exponentialcCr8)zêReturns an array of samples drawn from a standard gamma distribution. .. seealso:: - :func:`cupy.random.standard_gamma` for full documentation - :meth:`numpy.random.RandomState.standard_gamma` Nr+) r!r-r,r"rrBrrr)rr,rrr1rrrÚstandard_gammaMr<zRandomState.standard_gammacCs|j||dS)zäReturns samples drawn from the standard normal distribution. .. seealso:: - :func:`cupy.random.standard_normal` for full documentation - :meth:`numpy.random.RandomState.standard_normal` rŽ)re)rrrrrrr\szRandomState.standard_normalcCr8)zàReturns an array of samples drawn from the standard t distribution. .. seealso:: - :func:`cupy.random.standard_t` for full documentation - :meth:`numpy.random.RandomState.standard_t` Nr+) r!r-r,r"rZstandard_t_kernelrrrr9rrrÚ standard_tfr<zRandomState.standard_tcCsdddlm}|dur d}tj|tjd}|jjd}| |j|j j |j |¡|t  tj¡j M}|S)aîDraws integers between 0 and max integer inclusive. Return a sample of uniformly distributed random integers in the interval [0, ``np.iinfo(np.int_).max``]. The `np.int_` type translates to the C long integer type and its precision is platform dependent. Args: size (int or tuple of ints): Output shape. Returns: cupy.ndarray: Drawn samples. .. seealso:: :meth:`numpy.random.RandomState.tomaxint` rr Nrré)r r r!r"Úint_rÚitemsizer¼r r#r$rÚiinforµ)rrr r¸Z size_in_intrrrÚtomaxintus  ÿzRandomState.tomaxintzL left, M mode, R righta T base, leftbase, ratio, leftprod, rightprod; base = right - left; leftbase = mode - left; ratio = leftbase / base; leftprod = leftbase*base; rightprod = (right - mode)*base; if (x <= ratio) { x = left + sqrt(x*leftprod); } else { x = right - sqrt((1.0 - x) * rightprod); } Zcupy_triangular_kernelcCsšt |¡t |¡t |¡}}}t ||k¡rtdƒ‚t ||k¡r'tdƒ‚t ||k¡r2tdƒ‚|dur>t |||¡j}|j||d}t ||||¡S)a-Returns an array of samples drawn from the triangular distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.triangular` for full documentation - :meth:`numpy.random.RandomState.triangular` z left > modez mode > rightz left == rightNrŽ) r!r-rHrIr.r,r—rÚ_triangular_kernel)rÚleftÚmodeÚrightrrrMrrrÚ triangularªs ÿzRandomState.triangularz T low, T highzx = T(low) + x * T(high - low)Z cupy_scalecCsht |¡s t ||¡}t |¡st ||¡}|dur!t ||¡j}t |¡}|j||d}t  |||¡S)zÛReturns an array of uniformly-distributed samples over an interval. .. seealso:: - :func:`cupy.random.uniform` for full documentation - :meth:`numpy.random.RandomState.uniform` NrŽ) r¤r¥r!r-r.r,rr—rÚ _scale_kernel)rÚlowÚhighrrršrrrrÏÇs     zRandomState.uniformcCr*)zÛReturns an array of samples drawn from the von Mises distribution. .. seealso:: - :func:`cupy.random.vonmises` for full documentation - :meth:`numpy.random.RandomState.vonmises` Nr+) r!r-r.r,r"rZvonmises_kernelrrr)rÚmuÚkapparrr1rrrÚvonmisesÛr3zRandomState.vonmiseszT mean, T scale, T UzT Xa  T mu_2l; T Y; mu_2l = mean / (2*scale); Y = mean*X*X; X = mean + mu_2l*(Y - sqrt(4*scale*Y + Y*Y)); if (U > mean/(mean+X)) { X = mean*mean/X; } Zcupy_wald_scalecCs`tj||dtj||d}}|durt ||¡j}|j||d}|j||d}t ||||¡S)zÏReturns an array of samples drawn from the Wald distribution. .. seealso:: - :func:`cupy.random.wald` for full documentation - :meth:`numpy.random.RandomState.wald` rNrŽ)r!r-r.r,rer—rÚ _wald_kernel)rrirLrrrMrƒrrrÚwaldùsÿzRandomState.waldcCsRt |¡}t |dk¡rtdƒ‚|dur|j}| ||¡}tj|d||d|S)a$Returns an array of samples drawn from the weibull distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.weibull` for full documentation - :meth:`numpy.random.RandomState.weibull` rrNrFr\)r!r-rHrIr,rJrr‡rrrÚweibulls  zRandomState.weibullcCs\t |¡}t |dk¡rtdƒ‚|dur|j}tj||d}t ||j|¡|  |j ¡|S)aReturns an array of samples drawn from the Zipf distribution. .. warning:: This function may synchronize the device. .. seealso:: - :func:`cupy.random.zipf` for full documentation - :meth:`numpy.random.RandomState.zipf` rFz'a' must be a valid float > 1.0Nr+) r!r-rHrIr,r"rZ zipf_kernelrrr)rr/rrr1rrrÚzipfs  zRandomState.zipfTc Cs0|durtdƒ‚t|tjƒr|jdkrt‚t|tƒr%|}|dkr$tdƒ‚ntj|dd}|jdkr5tdƒ‚t|ƒ}|durpt |¡}|jdkrKtdƒ‚t|ƒ|krUtd ƒ‚|dk  ¡s_td ƒ‚t  |¡  ¡}t   |d¡sptd ƒ‚|durxtd ƒ‚|}t  |¡}|dkr‹|dkr‹td ƒ‚|s¸|dur¸||kr™tdƒ‚t|tƒr¦tj|dd}n| ¡}| |¡|d|… |¡S|s¼t‚|duræt |||f¡}tjt |¡|j||fddd} t|tƒsåt | |¡} n|dkrìd}|jd||d} | jtjdd} t|tƒr| S| jdkrtj|| |jdS|| S)zÃReturns an array of random values from a given 1-D array. .. seealso:: - :func:`cupy.random.choice` for full documentation - :meth:`numpy.random.choice` Nz%a must be 1-dimensional or an integerrz$a must be greater than or equal to 0Fr¢rzp must be 1-dimensionalza and p must have same sizez"probabilities are not non-negativezprobabilities do not sum to 1z5choice() without specifying size is not supported yetz-a cannot be empty unless no samples are takenz@Cannot take a larger sample than population when 'replace=False'Úlrr¾)r>)rIr?r!raÚndimrÉr@ÚarrayrtÚallrCÚgetr¤rxr Zaranger£Úshuffler%Z broadcast_toZargmaxr]ÚgumbelÚrandintr«r­r) rr/rÚreplacer6Za_sizeZp_sumr,rÁÚindexrrrÚchoice3s~ ÿ      ÿ  ÿ   ÿþ  €  zRandomState.choicecCsDt|tjƒs tdƒ‚|jdkrtdƒ‚|| t|ƒ¡|dd…<dS)z¥Returns a shuffled array. .. seealso:: - :func:`cupy.random.shuffle` for full documentation - :meth:`numpy.random.shuffle` zThe array must be cupy.ndarrayrz)An array whose ndim is 0 is not supportedN)r?r!rar”rëÚ _permutationrt©rr/rrrrïƒs  zRandomState.shufflecCs&t|tƒr | |¡S|| t|ƒ¡S)z6Returns a permuted range or a permutation of an array.)r?r@rõrtrörrrÚ permutation“s  zRandomState.permutationcCs@ddlm}tj|ftjd}| |j|jj |¡t  |¡}|S)zReturns a permuted range.rr r) r r r!r"r¤rªr¼r r#r$Zargsort)rr½r r¸rìrrrrõšs  zRandomState._permutationz$y = T(loc) - log(-log(x)) * T(scale)Zcupy_gumbel_kernelcCsdt |¡s t ||¡}t |¡st ||¡}|dur!t ||¡j}|j||d}t ||||¡|S)zÒReturns an array of samples drawn from a Gumbel distribution. .. seealso:: - :func:`cupy.random.gumbel` for full documentation - :meth:`numpy.random.RandomState.gumbel` NrŽ) r¤r¥r!r-r.r,rWrÚ_gumbel_kernelrYrrrrð¨s    zRandomState.gumbelc Cstt |¡s\t |¡}|durt |¡}|d}n |}t |¡d}|dur,t ||¡j}||}t t j |d¡}|  |  ¡|¡} |   |¡} t | |¡} |j |dd}tj| || d| S|durid}t|ƒd} n t|ƒ}t|ƒd} || kr{tdƒ‚|t |¡jkrŽtd t |¡j¡ƒ‚| t |¡jkr¡td  t |¡j¡ƒ‚| |}|  ||¡j |dd} tj| || d| S) zÛReturns a scalar or an array of integer values over ``[low, high)``. .. seealso:: - :func:`cupy.random.randint` for full documentation - :meth:`numpy.random.RandomState.randint` NrFr¢r\rz low >= highzlow is out of bounds for {}zhigh is out of bounds for {})r¤r¥r!r-Z zeros_liker.r,r®r¯r°r±rºÚflattenr«r%r_r@rIrÙÚminr§rÚnamerµ) rrárârrÚloÚhiÚdiffZ total_elemsr(Zhi1rMrrrrñ¹sP        ÿÿ ÿÿzRandomState.randint)NNr)NTN)DÚ__name__Ú __module__Ú __qualname__Ú__doc__rrr)r“r2r@r7r;rErNrQrRrTrUrZElementwiseKernelrXr[r`rkrlr„r†rerˆr‹rŒr…rršrœržrWr—rŸr¬r¶rºr²r´r¿rrÓrJrÔrrÕrÚrÛrßràrÏrårærçrèrérôrïr÷rõrørðrñrrrrrsÄ  ý ÿ`ÿ ýýc  ù %   íý ó P ýrcCstƒ |¡dS)a÷Resets the state of the random number generator with a seed. This function resets the state of the global random number generator for the current device. Be careful that generators for other devices are not affected. Args: seed (None or int): Seed for the random number generator. If ``None``, it uses :func:`os.urandom` if available or :func:`time.time` otherwise. Note that this function does not support seeding by an integer array. N)Úget_random_stater)rrrrrísrcCsiadSr)Ú_random_statesrrrrÚ reset_statessrcCsXt ¡}t |jd¡}|dur*t d¡}|durt t |ƒ¡}t |ƒ}t  |j|¡}|S)aXGets the state of the random number generator for the current device. If the state for the current device is not created yet, this function creates a new one, initializes it, and stores it as the state for the current device. Returns: RandomState: The state of the random number generator for the device. NZ CUPY_SEED) rZDevicerrîÚidrÇÚgetenvr¤r©r@rÚ setdefault)ÚdevÚrsrrrrr s  rcCs,t|tƒstd t|ƒ¡ƒ‚|tt ¡<dS)zSets the state of the random number generator for the current device. Args: state(RandomState): Random state to set for the current device. z;Random state must be an instance of RandomState. Actual: {}N)r?rr”r§ÚtyperrZ get_device_id)r rrrÚset_random_state s  þr cCs t |¡}|jdvrtdƒ‚|S)N)rQÚdz-cupy.random only supports float32 and float64)r¤rrhr”rrrrrg-s  rgr) ÚatexitrÅr®rËr°rÇrÊr r¤r}Z numpy.linalgrr!rrZ cupy.cudarZ cupy.randomrrrwr¨rÚobjectrrrÚregisterrrr rgrrrrÚsJ       Z