Fonction torch.div de PyTorch
Manuel de référence Pytorch torch
torch.divIl s'agit d'une fonction dans PyTorch utilisée pour effectuer une division élément par élément.
Définition de fonction
torch.div(input, other, rounding_mode, out)
Exemple d'utilisation
Exemple
import torch
a = torch.tensor([10, 20, 30])
b = torch.tensor([2, 4, 5])
# Division élément par élément
c = torch.div(a, b)
print(c)
a = torch.tensor([10, 20, 30])
b = torch.tensor([2, 4, 5])
# Division élément par élément
c = torch.div(a, b)
print(c)
Le résultat de sortie est:
tensor([5., 5., 6.])
Autres extensions