Fonction torch.divide de PyTorch
Manuel de référence PyTorch torch
torch.divideIl s'agit d'une fonction dans PyTorch pour la division élément par élément (équivalente à div).
Définition de fonction
torch.divide(input, other, *, rounding_mode=None, out=None)
Exemple d'utilisation
Exemple
import torch
# Exemple de division élément par élément
x = torch.tensor([10.0, 20.0, 30.0])
y = torch.tensor([2.0, 4.0, 5.0])
result = torch.divide(x, y)
print(result) # tensor([5.0000, 5.0000, 6.0000])
# Exemple de division élément par élément
x = torch.tensor([10.0, 20.0, 30.0])
y = torch.tensor([2.0, 4.0, 5.0])
result = torch.divide(x, y)
print(result) # tensor([5.0000, 5.0000, 6.0000])
Autres extensions