cupy.average#

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

返回沿某个轴的加权平均值。

参数:

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

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

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

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

  • keepdims (bool) – 如果 True，则轴将保持为大小为1的轴。

返回:

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

返回类型:

cupy.ndarray or tuple of cupy.ndarray

警告

如果给出 weight ，此函数可能会同步设备。

参见

numpy.average()