Функция abs
Функция abs вычисляет абсолютное значение (модуль) каждого элемента тензора. Она принимает тензор в качестве первого аргумента и возвращает новый тензор с результатами. Функция поддерживает работу с целочисленными и вещественными типами данных.
Синтаксис
torch.abs(tensor)
Пример
Вычислим абсолютные значения элементов одномерного тензора:
import torch
t = torch.tensor([-1, 2, -3, 4, -5])
res = torch.abs(t)
print(res)
Результат выполнения кода:
tensor([1, 2, 3, 4, 5])
Пример
Применим функцию к двумерному тензору с отрицательными числами:
import torch
t = torch.tensor([
[-1.5, 2.3, -3.7],
[4.1, -5.2, 6.8],
])
res = torch.abs(t)
print(res)
Результат выполнения кода:
tensor([
[1.5000, 2.3000, 3.7000],
[4.1000, 5.2000, 6.8000],
])
Пример
Функция abs также доступна как метод тензора. Используем её для обработки данных с зафиксированным зерном случайности:
import torch
torch.manual_seed(0)
t = torch.randn(3, 4) * 2
res = t.abs()
print(res)
Результат выполнения кода:
tensor([
[1.7641, 0.4002, 0.9787, 2.2409],
[1.8676, 0.9773, 0.9501, 0.1514],
[0.1032, 0.4106, 1.4543, 0.7610],
])
