Функция reciprocal
Функция reciprocal вычисляет обратные числа (1/x) для каждого элемента тензора.
Функция принимает тензор и возвращает новый тензор с обратными значениями.
Работает поэлементно, поддерживает автоматическое дифференцирование.
Синтаксис
torch.reciprocal(tensor)
Пример
Давайте вычислим обратные числа для элементов тензора:
import torch
t = torch.tensor([1.0, 2.0, 4.0, 10.0])
res = torch.reciprocal(t)
print(res)
Результат выполнения кода:
tensor([1.0000, 0.5000, 0.2500, 0.1000])
Пример
Функция работает и с целочисленными типами данных, но результат будет дробным:
import torch
t = torch.tensor([1, 2, 3, 4, 5])
res = torch.reciprocal(t)
print(res)
Результат выполнения кода:
tensor([1.0000, 0.5000, 0.3333, 0.2500, 0.2000])
Пример
Функция reciprocal поддерживает работу с многомерными тензорами:
import torch
t = torch.tensor([
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
])
res = torch.reciprocal(t)
print(res)
Результат выполнения кода:
tensor([
[1.0000, 0.5000, 0.3333],
[0.2500, 0.2000, 0.1667],
])
