o ְi7@sddlZddlZddlmZddlmZddlmZddlmZdddZ dd d Z dd d Z dd dZ dddZ    dddZddZ    dddddZ    dddddZddZdS)N)_core)_routines_statistics)_fusion_thread_local)contentFcC8tr|r tdtjtj|||dS|j|||dS)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.axisoutrr keepdims)r is_fusingNotImplementedErrorcall_reduction _statisticsaminminarr r rJ/home/ubuntu/.local/lib/python3.10/site-packages/cupy/_statistics/order.pyr  rcCr)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 r rramaxmaxrrrrr/rrcC0tj||||d}t|rtdt|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)rnanminrisnananywarningswarnRuntimeWarningrrr r resrrrrT rcCr)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)rnanmaxrrrrrr r!rrrr$sr#r$cCs|j|||dS)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 )ptprrrrr%sr%linearcs|jdkr |d}d}nd}|jdkrtd|jt|tr#|f}|rD|dur/dj}ntj}|D] } d|| j<q6t|}|durV|rO} n } d} nHtfdd|D}t t jt |} t | } t t| D] \} }| |qv|rjd| d } njd| d } d }| j|d | j|}||d }|d vrtd|d|dkrt|tj}n3|dkrt|tj}n%|dkrdt|t|}n|dkrtd|dkrntd||jtjkrt| |}|j|d|d}n>|dur*tj| jdd |jtjd}nt|d|j}tdddd|| | jdkrF| jd nd| j|t|d }|rZ|d}|rl|jdkrgd |}||}t !|S)NrTFz5Expected q to have a dimension of 1. Actual: {0} != 1)r'c3s|]}|jVqdS)N)ndim).0axrrr sz&_quantile_unchecked..)r-)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 enumeratesortedswapaxesreshapecopysortcupyfloorastypeint32ceilr=rollaxistakeemptyfloat64ElementwiseKernelsizesqueezer_internal_ascontiguousarray)rqrr overwrite_inputmethodr zerodkeepdimr*apnkeepkeepisNxindicesretrr+r_quantile_uncheckeds           ""     rkcCs$t|dkst|dkrdSdS)Ngr.FT)rQ count_nonzero)r^rrr_quantile_is_validsrm) interpolationc Cs`|dur t||d}t|tjstj|dd}t|d}t|s%tdt|||||||dS)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 percentiledr<dz)Percentiles must be in the range [0, 100]rr r_r`r ) _check_interpolation_as_methodrArQndarrayasarray true_dividermr?rkrr^rr r_r`r rnrrrro$s   roc CsT|dur t||d}t|tjstj|dd}t|stdt|||||||dS)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` Nquantilerpr<z%Quantiles must be in the range [0, 1]rr)rsrArQrtrurmr?rkrwrrrrxSs  rxcCs,tjd|dtdd|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`))rrDeprecationWarning TypeError)r`rnfnamerrrrss rs)NNF)NNFr&F)rrQr cupy._corerrr cupy._logicrrrrr$r%rkrmrorxrsrrrrs@      % %   m / /