Función torch.rsqrt de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.rsqrtEs la función de PyTorch utilizada para calcular la inversa de la raíz cuadrada elemento por elemento.

Definición de la función

torch.rsqrt(input, out=None)

Ejemplo de uso

Ejemplo

import torch

# Calcular el recíproco de la raíz cuadrada
x = torch.tensor([1.0, 4.0, 9.0])
result = torch.rsqrt(x)
print(result)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones