torcheval.metrics.functional.sum

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

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

参数：

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

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

引发：

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.)