Функция PyTorch torch.addcdiv

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

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

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

torch.addcdiv(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([4.0, 8.0, 12.0])
tensor2 = torch.tensor([2.0, 2.0, 2.0])

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

Результат вывода:

tensor([3., 6., 9.])

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

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