Fonction torch.abs de PyTorch
Manuel de référence Pytorch torch
torch.absC'est une fonction de PyTorch utilisée pour calculer la valeur absolue des tenseurs.
Définition de fonction
torch.abs(input, out=None)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur contenant des nombres négatifs
x = torch.tensor([-1, -2, 3, -4])
# Calculer les valeurs absolues
y = torch.abs(x)
print(y)
# Créer un tenseur contenant des nombres négatifs
x = torch.tensor([-1, -2, 3, -4])
# Calculer les valeurs absolues
y = torch.abs(x)
print(y)
Le résultat de sortie est:
tensor([1, 2, 3, 4])
Autres extensions