Fonction torch.addcdiv de PyTorch
Manuel de référence Pytorch torch
torch.addcdivest utilisé dans PyTorch pour effectuerinput + value * (tensor1 / tensor2)des opérations élément par élément.
Définition de fonction
torch.addcdiv(input, tensor1, tensor2, value=1, out=None)
Description du paramètre
input: tenseur d'entréetensor1: tenseur numérateurtensor2: tenseur dénominateur (ne peut pas être zéro)value: multiplicateur, la valeur par défaut est 1out: tenseur de sortie (optionnel)
Exemple d'utilisation
Exemple
import torch
# Créer des tenseurs
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([4.0, 8.0, 12.0])
tensor2 = torch.tensor([2.0, 2.0, 2.0])
# Effectuer input + value * (tensor1 / tensor2)
result = torch.addcdiv(input, tensor1, tensor2, value=1)
print(result)
# Créer des tenseurs
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([4.0, 8.0, 12.0])
tensor2 = torch.tensor([2.0, 2.0, 2.0])
# Effectuer input + value * (tensor1 / tensor2)
result = torch.addcdiv(input, tensor1, tensor2, value=1)
print(result)
Le résultat de sortie est:
tensor([3., 6., 9.])
Autres extensions