torch.addcmul

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

Выполняет поэлементное умножение tensor1 на tensor2, умножает результат на скаляр value и добавляет его к input.

$$\text{out}_i = \text{input}_i + \text{value} \times \text{tensor1}_i \times \text{tensor2}_i$$

Формы tensor, tensor1, и tensor2 должны быть совместимы для трансляции.

Для входных данных типа FloatTensor или DoubleTensor, value должно быть действительным числом, в противном случае целым.

Параметры:

• input (Tensor) – тензор, который будет добавлен
• tensor1 (Tensor) – тензор, который будет умножен
• tensor2 (Tensor) – тензор, который будет умножен

Ключевые аргументы:

• value (Число, необязательно) – множитель для $tensor1 .* tensor2$
• out (Tensor, необязательно) – выходной тензор.

Пример:

>>> t = torch.randn(1, 3)
>>> t1 = torch.randn(3, 1)
>>> t2 = torch.randn(1, 3)
>>> torch.addcmul(t, t1, t2, value=0.1)
tensor([[-0.8635, -0.6391,  1.6174],
        [-0.7617, -0.5879,  1.7388],
        [-0.8353, -0.6249,  1.6511]])