Функция tanh
Функция torch.tanh применяется к тензору поэлементно,
возвращая новый тензор с вычисленными значениями гиперболического
тангенса для каждого входного элемента. Первым параметром
функция принимает входной тензор. Также можно указать
выходной тензор через параметр out.
Синтаксис
torch.tanh(input, *, out=None)
Пример
Давайте применим функцию tanh к одномерному тензору:
import torch
t = torch.tensor([-1.0, 0.0, 1.0])
res = torch.tanh(t)
print(res)
Результат выполнения кода:
tensor([-0.7616, 0.0000, 0.7616])
Пример
Применим функцию tanh к двумерному тензору:
import torch
t = torch.tensor([
[0.5, 1.0],
[2.0, 3.0],
])
res = torch.tanh(t)
print(res)
Результат выполнения кода:
tensor([
[0.4621, 0.7616],
[0.9640, 0.9951],
])
Пример
Используем функцию tanh вместе с методом tanh тензора:
import torch
t = torch.tensor([-2.0, -1.0, 0.0, 1.0, 2.0])
res_func = torch.tanh(t)
res_method = t.tanh()
print(res_func)
print(res_method)
Результат выполнения кода:
tensor([-0.9640, -0.7616, 0.0000, 0.7616, 0.9640])
tensor([-0.9640, -0.7616, 0.0000, 0.7616, 0.9640])
