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) – 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()