cupy.average#

cupy.average(a, axis=None, weights=None, returned=False, *, keepdims=False)[source]#

返回沿轴加权平均值。

参数:

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

  • axis (int) – 计算平均值的轴。默认使用展平的数组。

  • weights (cupy.ndarray) – 权重数组，其中每个元素对应 a 中的值。如果为 None，则 a 中的所有值权重都等于一。

  • returned (bool) – 如果为 True，则返回平均值和权重总和的元组；否则仅返回平均值。

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

返回值:

输入数组沿轴的平均值和权重总和。

返回类型:

cupy.ndarray 或 tuple of cupy.ndarray

警告

如果指定了 weight，此函数可能会同步设备。

另请参阅

numpy.average()