torch.abs

torch.abs(input, *, out=None) → 张量

计算input中每个元素的绝对值。

$$\text{out}_{i} = |\text{input}_{i}|$$

Parameters

输入 (张量) – 输入张量。

Keyword Arguments

输出 (张量, 可选) – 输出张量。

示例：

>>> torch.abs(torch.tensor([-1, -2, 3]))
tensor([ 1,  2,  3])