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 的形狀必須是 可廣播的 (broadcastable)。

對於 FloatTensor 或 DoubleTensor 類型的輸入，value 必須是實數，否則必須是整數。

參數

• input (Tensor) – 要被加上的張量

• tensor1 (Tensor) – 要被乘上的張量

• tensor2 (Tensor) – 要被乘上的張量

關鍵字參數

• value (Number, optional) – $tensor1 .* tensor2$ 的乘數

• out (Tensor, optional) – 輸出張量。

範例

>>> 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]])