cupy.mean

cupy.mean(a, axis=None, dtype=None, out=None, keepdims=False)

返回沿轴的算术平均值。

参数:

• a (cupy.ndarray) – 要计算平均值的数组。

• axis (int, int 序列 或 None) – 计算平均值时沿哪个轴。默认使用展平的数组。

• dtype – 数据类型说明符。

• out (cupy.ndarray) – 输出数组。

• keepdims (bool) – 如果为 True，则轴保留为大小为一的轴。

返回:

输入数组沿轴的平均值。

返回类型:

cupy.ndarray

另请参阅

numpy.mean()