Fonction torch.pow de PyTorch
Manuel de référence Pytorch torch
torch.powIl s'agit d'une fonction de PyTorch utilisée pour calculer des opérations de puissance élément par élément.
Définition de fonction
torch.pow(input, exponent, out)
Exemple d'utilisation
Exemple
import torch
# Calculer l'opération de puissance
x = torch.tensor([2, 3, 4])
result = torch.pow(x, 2)
print(result)
# Vous pouvez également utiliser l'exposant comme premier argument
result2 = torch.pow(2, x)
print(result2)
# Calculer l'opération de puissance
x = torch.tensor([2, 3, 4])
result = torch.pow(x, 2)
print(result)
# Vous pouvez également utiliser l'exposant comme premier argument
result2 = torch.pow(2, x)
print(result2)
Autres extensions