Функция torch.sqrt в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.sqrt— это функция в PyTorch, используемая для поэлементного вычисления квадратного корня тензоров.

Определение функции

torch.sqrt(input, out=None)

Примеры использования

Пример

import torch

x = torch.tensor([4.0, 9.0, 16.0, 25.0])

y = torch.sqrt(x)
print(y)

Результат вывода:

tensor([2., 3., 4., 5.])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения