PyTorch torch.addcmul-Funktion
PyTorch torch Referenzhandbuch
torch.addcmulIst eine Funktion in PyTorch, die verwendet wird, um auszuführeninput + value * (tensor1 * tensor2)elementweise Operationen.
Funktionsdefinition
torch.addcmul(input, tensor1, tensor2, value=1, out=None)
Parameterbeschreibung
input: Eingabetensortensor1: erster Multiplikator-Tensortensor2: zweiter Multiplikator-Tensorvalue: Multiplikator, Standardwert ist 1out: Ausgabe-Tensor (optional)
Anwendungsbeispiel
Beispiel
import torch
# Tensor erstellen
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([2.0, 3.0, 4.0])
tensor2 = torch.tensor([3.0, 4.0, 5.0])
# Führe input + value * (tensor1 * tensor2) aus
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)
# Tensor erstellen
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([2.0, 3.0, 4.0])
tensor2 = torch.tensor([3.0, 4.0, 5.0])
# Führe input + value * (tensor1 * tensor2) aus
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)
Die Ausgabe ist:
tensor([ 7., 14., 23.])
Andere Erweiterungen