cupy.mean#

cupy.mean(a, axis=None, dtype=None, out=None, keepdims=False)[source]#

Returns the arithmetic mean along an axis.

Parameters:
  • 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:

The mean of the input array along the axis.

Return type:

cupy.ndarray

See also

numpy.mean()