Функция PyTorch torch.addcmul

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.addcmulЯвляется функцией в PyTorch, используемой для выполненияinput + value * (tensor1 * tensor2)поэлементных операций.

Определение функции

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

Описание параметра

  • input: входной тензор
  • tensor1: первый тензор-множитель
  • tensor2: второй тензор-множитель
  • value: множитель, по умолчанию 1
  • out: выходной тензор (необязательно)

Пример использования

Пример

import torch

# Создать тензор
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])

# Выполнить input + value * (tensor1 * tensor2)
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)

Вывод:

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

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения