o X۷i@sDddlmZddlZddlZddlmZddZddZdd ZdS) ) annotationsN)specialcCstj|ddtt|S)Nraxis)r logsumexpmathloglen)logxr R/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/stats/_morestats.py _log_mean sr cCsVtt|tjd|j}tjt||gdd}tjd|ddjt t |S)Ny?rr) cupy broadcast_tor pishaperrasarrayrealrrr )r neg_logmeanlogxmur r r _log_vars"rc Cs|jjdvr |tj}|j}|jtjkr|tj}|jd}|dkr,tjtj |dSt |}|dkrBtj |dd}t |}n||}t |dt t|}|dtj|dd|d|}|j|ddS) aThe boxcox log-likelihood function. Parameters ---------- lmb : scalar Parameter for Box-Cox transformation data : array-like Data to calculate Box-Cox log-likelihood for. If `data` is multi-dimensional, the log-likelihood is calculated along the first axis Returns ------- llf : float or cupy.ndarray Box-Cox log-likelihood of `data` given `lmb`. A float for 1-D `data`, an array otherwise See Also -------- scipy.stats.boxcox_llf biur)dtyperrF)copy)rkindastyperfloat64float16float32rarraynanrvarrrabssum) lmbdatarNlogdatavariancelogvarr resr r r boxcox_llfs       "r-) __future__rrr cupyx.scipyrr rr-r r r r s