torcheval.metrics.functional.sum

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

计算加权和。当未提供权重时，它计算未加权和。 其类版本是 torcheval.metrics.Sum。

Parameters:

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

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

Raises:

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

示例：

>>> import torch
>>> from torcheval.metrics.functional import sum
>>> sum(torch.tensor([2, 3]))
tensor(5.)
>>> sum(torch.tensor([2, 3]), torch.tensor([0.1, 0.6]))
tensor(2.)
>>> sum(torch.tensor([2, 3]), 0.5)
tensor(2.5)
>>> sum(torch.tensor([2, 3]), 2)
tensor(10.)