torcheval.metrics.functional.mean

torcheval.metrics.functional.mean(input: Tensor, weight: float | int | Tensor = 1.0) → Tensor

计算加权平均值。当未提供权重时，它计算未加权的平均值。 其类版本是 torcheval.metrics.Mean。

加权平均值 = sum(weight * input) / sum(weight)

Parameters:

• input (Tensor) – 输入值的张量。

• weight (可选) – 输入权重的浮点数、整数或张量。默认值为1.0。如果weight是张量，其大小应与输入张量的大小匹配。

Raises:

ValueError – 如果权重的值既不是float，也不是int，也不是与输入张量大小匹配的torch.Tensor。

示例：

>>> import torch
>>> from torcheval.metrics.functional import mean
>>> mean(torch.tensor([2, 3]))
tensor(2.5)
>>> mean(torch.tensor([2, 3]), torch.tensor([0.2, 0.8]))
tensor(2.8)
>>> mean(torch.tensor([2, 3]), 0.5)
tensor(2.5)
>>> mean(torch.tensor([2, 3]), 1)
tensor(2.5)