`i7ddlZddlZddlmZddlmZddlmZddlmZddZ ddZ dd Z dd Z dd Z dd ZdZ ddddZ ddddZdZdS)N)_core)_routines_statistics)_fusion_thread_local)contentFctjr3|rtdtjtj|||S||||S)aReturns the minimum of an array or the minimum along an axis. .. note:: When at least one element is NaN, the corresponding min value will be NaN. Args: a (cupy.ndarray): Array to take the minimum. axis (int): Along which axis to take the minimum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The minimum of ``a``, along the axis if specified. .. note:: When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements. .. seealso:: :func:`numpy.amin` z4cupy.amin does not support `keepdims` in fusion yet.axisoutr r keepdims)r is_fusingNotImplementedErrorcall_reduction _statisticsaminminar r r s j/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_statistics/order.pyrr v4%''5  H%FHH H#2  ad555 5 55dh5 7 77ctjr3|rtdtjtj|||S||||S)aReturns the maximum of an array or the maximum along an axis. .. note:: When at least one element is NaN, the corresponding min value will be NaN. Args: a (cupy.ndarray): Array to take the maximum. axis (int): Along which axis to take the maximum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The maximum of ``a``, along the axis if specified. .. note:: When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements. .. seealso:: :func:`numpy.amax` z4cupy.amax does not support `keepdims` in fusion yet.rr )rr rrramaxmaxrs rrr/rrctj||||}tj|rt jdt|S)aReturns the minimum of an array along an axis ignoring NaN. When there is a slice whose elements are all NaN, a :class:`RuntimeWarning` is raised and NaN is returned. Args: a (cupy.ndarray): Array to take the minimum. axis (int): Along which axis to take the minimum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The minimum of ``a``, along the axis if specified. .. warning:: This function may synchronize the device. .. seealso:: :func:`numpy.nanmin` r All-NaN slice encountered)rnanminrisnananywarningswarnRuntimeWarningrr r r ress rrrTS2 ,qtx @ @ @C}SC 1>BBB Jrctj||||}tj|rt jdt|S)aReturns the maximum of an array along an axis ignoring NaN. When there is a slice whose elements are all NaN, a :class:`RuntimeWarning` is raised and NaN is returned. Args: a (cupy.ndarray): Array to take the maximum. axis (int): Along which axis to take the maximum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The maximum of ``a``, along the axis if specified. .. warning:: This function may synchronize the device. .. seealso:: :func:`numpy.nanmax` r r)rnanmaxrrrr r!r"r#s rr'r'sr%rc2||||S)aPReturns the range of values (maximum - minimum) along an axis. .. note:: The name of the function comes from the acronym for 'peak to peak'. When at least one element is NaN, the corresponding ptp value will be NaN. Args: a (cupy.ndarray): Array over which to take the range. axis (int): Axis along which to take the minimum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is retained as an axis of size one. Returns: cupy.ndarray: The minimum of ``a``, along the axis if specified. .. note:: When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements. .. seealso:: :func:`numpy.amin` r )ptprs rr)r)s8 55dh5 7 77rlinearcZ|jdkr |d}d}nd}|jdkr'td|jt|tr|f}|rB| djz}n5t j}|D]} d|| jz<t|}|.|r} n } d} ntfd|D}ttjt|z } t| } tt| D]\} }| ||r&jd| dz} n7jd| dz} d }| | | j|}||d z z}|d vrtd |d|dkr2t'j|t&j}n|dkr2t'j|t&j}nq|dkr-dt'j|t'j|zz}n>|dkrtd|dkrn"td||jt&jkr.t'j| |}||d|}n|6t'j| jdd |jzt&j}nt'j|d|j}t'jdddd|| | jdkr | jd nd| j|t'j|d }|r|d}|r%|jdkrd|z}||}tAj!|S)NrTFz5Expected q to have a dimension of 1. Actual: {0} != 1)r,c3,K|]}|jzVdS)N)ndim).0axrs r z&_quantile_unchecked..s)00RR!&[000000r)r2)r ?) inverted_cdfaveraged_inverted_cdfclosest_observationinterpolated_inverted_cdfhazenweibullmedian_unbiasednormal_unbiased'z;' method is not yet supported. Please use any other method.lowerhighermidpointg?nearestzC'nearest' method is not yet supported. Please use any other method.r*z_Unexpected interpolation method. Actual: '{0}' not in ('linear', 'lower', 'higher', 'midpoint')rdtypez0S idx, raw T a, raw int32 offset, raw int32 sizezU reta# ptrdiff_t idx_below = floor(idx); U weight_above = idx - idx_below; ptrdiff_t max_idx = size - 1; ptrdiff_t offset_bottom = _ind.get()[0] * offset + idx_below; ptrdiff_t offset_top = min(offset_bottom + 1, max_idx); U diff = a[offset_top] - a[offset_bottom]; if (weight_above < 0.5) { ret = a[offset_bottom] + diff * weight_above; } else { ret = a[offset_top] - diff * (1 - weight_above); } cupy_percentile_weightnening)"r. ValueErrorformat isinstanceintlistshapetupleravelflattensetrangelen enumeratesortedswapaxesreshapecopysortcupyfloorastypeint32ceilrBrollaxistakeemptyfloat64ElementwiseKernelsizesqueezer_internal_ascontiguousarray)rqr r overwrite_inputmethodr zerodkeepdimr0apnkeepkeepisNxindicesrets` r_quantile_uncheckedrpsW v{{ dGvzz,,2F16NN<< <$u% <QVmGG17mmG ) )'(QV $$GnnG |  BBB00004000005==!!CII-D fTll++ ! !DAq 1a  AA  ;176E6?U233BB176E6?U23388::B DGGG $B27mG LLL8f88899 9 7  *W%%,,TZ88 8  )G$$++DJ77 :  G,,ty/A/AAB 9  899 9 8   55;VF^^EE E} ""mB%%hhwQCh00 ;*RXcrc]QW4DLIIICC-Q11C  >  +%  & 2rw{{rx||27C' I I I(mC$$ kk!nn# 6A::goGkk'""  ,S 1 11rcjtj|dkstj|dkrdSdS)Ngr3FT)rV count_nonzero)rcs r_quantile_is_validrss9 !c'""d&8S&A&Au 4r) interpolationc |t||d}t|tjstj|d}tj|d}t |stdt|||||||S)a Computes the q-th percentile of the data along the specified axis. Args: a (cupy.ndarray): Array for which to compute percentiles. q (float, tuple of floats or cupy.ndarray): Percentiles to compute in the range between 0 and 100 inclusive. axis (int or tuple of ints): Along which axis or axes to compute the percentiles. The flattened array is used by default. out (cupy.ndarray): Output array. overwrite_input (bool): If True, then allow the input array `a` to be modified by the intermediate calculations, to save memory. In this case, the contents of the input `a` after this function completes is undefined. method (str): Interpolation method when a quantile lies between two data points. ``linear`` interpolation is used by default. Supported interpolations are``lower``, ``higher``, ``midpoint``, ``nearest`` and ``linear``. keepdims (bool): If ``True``, the axis is remained as an axis of size one. interpolation (str): Deprecated name for the method keyword argument. Returns: cupy.ndarray: The percentiles of ``a``, along the axis if specified. .. seealso:: :func:`numpy.percentile` N percentiledrAdz)Percentiles must be in the range [0, 100]r r rdrer ) _check_interpolation_as_methodrFrVndarrayasarray true_dividersrDrprrcr r rdrer rts rrvrv$s@ / M<11 a & &' L# & & & C  A a FDEEE  14S'    rc |t||d}t|tjstj|d}t |st dt|||||||S)aComputes the q-th quantile of the data along the specified axis. Args: a (cupy.ndarray): Array for which to compute quantiles. q (float, tuple of floats or cupy.ndarray): Quantiles to compute in the range between 0 and 1 inclusive. axis (int or tuple of ints): Along which axis or axes to compute the quantiles. The flattened array is used by default. out (cupy.ndarray): Output array. overwrite_input (bool): If True, then allow the input array `a` to be modified by the intermediate calculations, to save memory. In this case, the contents of the input `a` after this function completes is undefined. method (str): Interpolation method when a quantile lies between two data points. ``linear`` interpolation is used by default. Supported interpolations are``lower``, ``higher``, ``midpoint``, ``nearest`` and ``linear``. keepdims (bool): If ``True``, the axis is remained as an axis of size one. interpolation (str): Deprecated name for the method keyword argument. Returns: cupy.ndarray: The quantiles of ``a``, along the axis if specified. .. seealso:: :func:`numpy.quantile` NquantilerwrAz%Quantiles must be in the range [0, 1]ry)rzrFrVr{r|rsrDrpr~s rrrSs@ / M:// a & &' L# & & & a B@AAA  14S'    rcptjd|dtd|dkrtd|S)Nz!the `interpolation=` argument to z was renamed to `method=`, which has additional options. Users of the modes 'nearest', 'lower', 'higher', or 'midpoint' are encouraged to review the method they. (Deprecated NumPy 1.22)) stacklevelr*zjYou shall not pass both `method` and `interpolation`! (`interpolation` is Deprecated in favor of `method`))r r!DeprecationWarning TypeError)rertfnames rrzrzsh M "E " " " q **** CDD D r)NNF)NNFr*F)r rVr cupy._corerrr cupy._logicrrrrr'r)rprsrvrrzrrrsx ::::::++++++"8"8"8"8J"8"8"8"8J>>8888>.2(-'!&j2j2j2j2Z %)$, " ,,,,,^#'"+  +++++^r