Fonction torch.exp2 de PyTorch
Manuel de référence Pytorch torch
torch.exp2C'est une fonction de PyTorch utilisée pour calculer la puissance de 2 élément par élément.
Définition de fonction
torch.exp2(input, *, out=None)
Exemple d'utilisation
Exemple
import torch
# Exemple de puissance de 2
x = torch.tensor([0, 1, 2, 3, 4], dtype=torch.float32)
result = torch.exp2(x)
print(result) # tensor([1., 2., 4., 8., 16.])
# Exemple de puissance de 2
x = torch.tensor([0, 1, 2, 3, 4], dtype=torch.float32)
result = torch.exp2(x)
print(result) # tensor([1., 2., 4., 8., 16.])
Autres extensions