Fonction torch.addcmul de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.addcmulEst une fonction dans PyTorch utilisée pour effectuerinput + value * (tensor1 * tensor2)des opérations élément par élément.

Définition de fonction

torch.addcmul(input, tensor1, tensor2, value=1, out=None)

Description du paramètre

  • input: tenseur d'entrée
  • tensor1: premier tenseur multiplicateur
  • tensor2: deuxième tenseur multiplicateur
  • value: multiplicateur, valeur par défaut 1
  • out: tenseur de sortie (optionnel)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur
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])

# Exécuter input + value * (tensor1 * tensor2)
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)

La sortie est :

tensor([ 7., 14., 23.])

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions