Fonction torch.sqrt de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.sqrtest une fonction de PyTorch utilisée pour calculer la racine carrée élément par élément des tenseurs.

Définition de fonction

torch.sqrt(input, out=None)

Exemples d'utilisation

Exemple

import torch

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

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

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions