Функция atan
Функция atan вычисляет арктангенс для каждого элемента тензора. Результат возвращается в радианах в диапазоне от -π/2 до π/2. Функция принимает на вход тензор любого размера и применяет операцию поэлементно. Поддерживаются все числовые типы данных, включая целочисленные и числа с плавающей точкой.
Синтаксис
torch.atan(input, *, out=None)
Пример
Давайте вычислим арктангенс для чисел из списка:
import torch
t = torch.tensor([0, 1, -1, 0.5, -0.5])
res = torch.atan(t)
print(res)
Результат выполнения кода:
tensor([ 0.0000, 0.7854, -0.7854, 0.4636, -0.4636])
Пример
Давайте применим функцию к двумерному тензору:
import torch
t = torch.tensor([[1, 2, 3], [4, 5, 6]])
res = torch.atan(t)
print(res)
Результат выполнения кода:
tensor([
[0.7854, 1.1071, 1.2490],
[1.3258, 1.3734, 1.4056],
])
Пример
Давайте используем аргумент out для записи результата в заранее созданный тензор:
import torch
t = torch.tensor([1, 0, -1])
out_t = torch.empty(3)
torch.atan(t, out=out_t)
print(out_t)
Результат выполнения кода:
tensor([ 0.7854, 0.0000, -0.7854])
