`i(xddlZddlZddlZddlmZddZddZddddZddZ dd Z dd Z dd Z dd Z dd ZdS)N)_routines_statisticsFc2tj|||||S)a?Compute the median along the specified axis. Returns the median of the array elements. Args: a (cupy.ndarray): Array to compute the median. axis (int, sequence of int or None): Axis along which the medians are computed. The flattened array is used by default. out (cupy.ndarray): Output array. overwrite_input (bool): If ``True``, then allow use of memory of input array a for calculations. The input array will be modified by the call to median. This will save memory when you do not need to preserve the contents of the input array. Treat the input as undefined, but it will probably be fully or partially sorted. Default is ``False``. If ``overwrite_input`` is ``True`` and ``a`` is not already an ndarray, an error will be raised. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The median of ``a``, along the axis if specified. .. seealso:: :func:`numpy.median` ) _statistics_medianaaxisoutoverwrite_inputkeepdimss l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_statistics/meanvar.pymedianrs4  q$_h G GGcv|jjdvrtj|||||St |||||S)aVCompute the median along the specified axis, while ignoring NaNs. Returns the median of the array elements. Args: a (cupy.ndarray): Array to compute the median. axis (int, sequence of int or None): Axis along which the medians are computed. The flattened array is used by default. out (cupy.ndarray): Output array. overwrite_input (bool): If ``True``, then allow use of memory of input array a for calculations. The input array will be modified by the call to median. This will save memory when you do not need to preserve the contents of the input array. Treat the input as undefined, but it will probably be fully or partially sorted. Default is ``False``. If ``overwrite_input`` is ``True`` and ``a`` is not already an ndarray, an error will be raised. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The median of ``a``, along the axis if specified. .. seealso:: :func:`numpy.nanmedian` efdFD)r r r r )dtypecharr _nanmedianrrs r nanmedianr%sN4 w|w%asOXNNNad_'))) )rr ctj|}|@|||}|j|j|jz }ntj|}t |jjtjtj fr-tj tj |j|jdf}ntj |j|j}|j |j kr|td|jdkrtd|j d|j |krt!dtj||jdz d z|j z}|d |}|||| }tj|d krt+d tj||||||z }|rR|j |j kr>tjtj||j }||fS|S)aReturns the weighted average along an axis. Args: a (cupy.ndarray): Array to compute average. axis (int): Along which axis to compute average. The flattened array is used by default. weights (cupy.ndarray): Array of weights where each element corresponds to the value in ``a``. If ``None``, all the values in ``a`` have a weight equal to one. returned (bool): If ``True``, a tuple of the average and the sum of weights is returned, otherwise only the average is returned. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray or tuple of cupy.ndarray: The average of the input array along the axis and the sum of weights. .. warning:: This function may synchronize the device if ``weight`` is given. .. seealso:: :func:`numpy.average` N)r r f8z;Axis must be specified when shapes of a and weights differ.z81D weights expected when shapes of a and weights differ.rz5Length of weights not compatible with specified axis.)r)r rr gz(Weights sum to zero, can't be normalized)rr)cupyasarraymeanrtypesize issubclassnumpyintegerbool_ functoolsreduce promote_typesshape TypeErrorndim ValueError broadcast_toswapaxessumanyZeroDivisionErrormultiplyarraycopy) rr weightsreturnedr avgsclwgt result_dtypes r averager9Fs>4 QAff$f22innQVch.//l7## aglU]EK$@ A A C$+E,?-.Wci,FHHLL!.qw BBL 7ci  |x1}}NPPPy|qwt},, KMMM#C!&1*))stdrAs r rDrDrBrc|jjdvr|||||Stj|||||S)a6Returns the arithmetic mean along an axis ignoring NaN values. Args: a (cupy.ndarray): Array to compute mean. axis (int, sequence of int or None): Along which axis to compute mean. The flattened array is used by default. dtype: Data type specifier. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The mean of the input array along the axis ignoring NaNs. .. seealso:: :func:`numpy.nanmean` biur;)rkindrr_nanmeanr<s r nanmeanrIsT$ w|uvv4u#vIII   EsX ? ? ??rc|jjdvr||||||Stj||||||S)aReturns the variance along an axis ignoring NaN values. Args: a (cupy.ndarray): Array to compute variance. axis (int): Along which axis to compute variance. The flattened array is used by default. dtype: Data type specifier. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The variance of the input array along the axis. .. seealso:: :func:`numpy.nanvar` rFr>)rrGr@r_nanvarrAs r nanvarrLf$ w|uuu$e4&(( (   Es J J JJrc|jjdvr||||||Stj||||||S)a;Returns the standard deviation along an axis ignoring NaN values. Args: a (cupy.ndarray): Array to compute standard deviation. axis (int): Along which axis to compute standard deviation. The flattened array is used by default. dtype: Data type specifier. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The standard deviation of the input array along the axis. .. seealso:: :func:`numpy.nanstd` rFr>)rrGrDr_nanstdrAs r nanstdrPrMr)NNFF)NNF)NNNF)NNNrF)r$r!r cupy._corerrrrr9rr@rDrIrLrPrr rSs ::::::HHHH:))))BFEFFFFFRFFFF,$$$$.$$$$.????4JJJJ6JJJJJJr