Функция tan
Функция tan вычисляет тангенс для каждого элемента тензора.
На вход подается тензор произвольной формы.
На выходе получается тензор той же формы,
где каждый элемент заменен на значение тангенса от исходного значения.
Углы задаются в радианах.
Функция поддерживает автоматическое дифференцирование.
Синтаксис
torch.tan(input_tensor)
Пример
Давайте вычислим тангенс для чисел 0, 0.5, 1, 1.5:
import torch
t = torch.tensor([0, 0.5, 1, 1.5])
res = torch.tan(t)
print(res)
Результат выполнения кода:
tensor([0.0000, 0.5463, 1.5574, 14.1014])
Пример
Давайте передадим функции двумерный тензор:
import torch
t = torch.tensor([
[0, 0.25],
[0.5, 0.75],
])
res = torch.tan(t)
print(res)
Результат выполнения кода:
tensor([
[0.0000, 0.2553],
[0.5463, 0.9316],
])
Пример
Используем функцию в градиентной операции:
import torch
t = torch.tensor([0.5, 1.0], requires_grad=True)
res = torch.tan(t)
loss = res.sum()
loss.backward()
print(t.grad)
Результат выполнения кода:
tensor([1.2984, 3.4255])