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)

参数：

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

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

引发：

ValueError – 如果 weight 的值既不是 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)