Функция sqrt
Функция sqrt вычисляет квадратный корень для каждого элемента тензора.
На вход она принимает тензор, а возвращает новый тензор того же размера,
где каждое значение заменено на его квадратный корень.
Функция поддерживает работу с целочисленными и вещественными типами данных,
при этом целочисленные тензоры автоматически преобразуются в тип с плавающей запятой.
Для отрицательных значений возвращается nan.
Синтаксис
torch.sqrt(tensor)
Пример
Давайте вычислим квадратный корень из чисел от 1 до 5:
import torch
t = torch.tensor([1, 4, 9, 16, 25])
res = torch.sqrt(t)
print(res)
Результат выполнения кода:
tensor([1., 2., 3., 4., 5.])
Пример
Если передать целочисленный тензор, результат будет преобразован в вещественный тип:
import torch
t = torch.tensor([0, 1, 2, 3, 4], dtype=torch.int)
res = torch.sqrt(t)
print(res)
Результат выполнения кода:
tensor([0.0000, 1.0000, 1.4142, 1.7321, 2.0000])
Пример
Функцию можно вызывать как метод тензора:
import torch
t = torch.tensor([1.0, 4.0, 9.0])
res = t.sqrt()
print(res)
Результат выполнения кода:
tensor([1., 2., 3.])
