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

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

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

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

torch.rsqrt(input, out=None)

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

Пример

import torch

# Вычислить обратную величину квадратного корня
x = torch.tensor([1.0, 4.0, 9.0])
result = torch.rsqrt(x)
print(result)

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

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