Fonction torch.rsqrt de PyTorch
Manuel de référence Pytorch torch
torch.rsqrtIl s'agit d'une fonction de PyTorch utilisée pour calculer l'inverse élément par élément de la racine carrée.
Définition de fonction
torch.rsqrt(input, out=None)
Exemple d'utilisation
Exemple
import torch
# Calculer l'inverse de la racine carrée
x = torch.tensor([1.0, 4.0, 9.0])
result = torch.rsqrt(x)
print(result)
# Calculer l'inverse de la racine carrée
x = torch.tensor([1.0, 4.0, 9.0])
result = torch.rsqrt(x)
print(result)
Autres extensions