Función PyTorch torch.addcmul

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.addcmulEs la función de PyTorch que se utiliza para realizarinput + value * (tensor1 * tensor2)operaciones elemento por elemento.

Definición de la función

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

Descripción de parámetros

  • input: tensor de entrada
  • tensor1: primer tensor multiplicador
  • tensor2: segundo tensor multiplicador
  • value: multiplicador, el valor predeterminado es 1
  • out: tensor de salida (opcional)

Ejemplo de uso

Ejemplo

import torch

# Crear tensores
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])

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

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones