torch.abs

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

Вычисляет абсолютное значение каждого элемента в input.

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

Параметры:

input (Tensor) – входной тензор.

Ключевые аргументы:

out (Tensor, необязательно) – выходной тензор.

Пример:

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