torch.abs

torch.abs(input: Tensor, *, out: Optional[Tensor]) → Tensor

計算 input 中每個元素的絕對值。

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

引數

input (Tensor) – 輸入張量。

關鍵字引數

out (Tensor, optional) – 輸出張量。

示例

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