cupy.average¶
-
cupy.
average
(a, axis=None, weights=None, returned=False)[source]¶ Returns the weighted average along an axis.
Parameters: - 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
. IfNone
, all the values ina
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.
Returns: - The average of the input array
along the axis and the sum of weights.
Return type: cupy.ndarray or tuple of cupy.ndarray
See also