Python PyTorch addcdiv用法及代碼示例

本文簡要介紹python語言中 torch.addcdiv 的用法。

用法:

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

參數：

• input(Tensor) -要添加的張量

• tensor1(Tensor) -分子張量

• tensor2(Tensor) -分母張量

關鍵字參數：

• value(數字,可選的) -\text{tensor1} / \text{tensor2} 的乘數

• out(Tensor,可選的) -輸出張量。

執行 tensor1 除以 tensor2 的元素除法，將結果乘以標量 value 並將其添加到 input 。

警告

不再支持使用 addcdiv 進行整數除法，在未來的版本中，addcdiv 將執行 tensor1 和 tensor2 的真正除法。曆史 addcdiv 行為可以實現為 (input + value * torch.trunc(tensor1 /tensor2)).to(input.dtype) 用於整數輸入和 (input + value * tensor1 /tensor2) 用於浮點輸入。未來的 addcdiv 行為隻是後一種實現：(input + value * tensor1 /tensor2)，適用於所有 dtype。

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

input 、 tensor1 和 tensor2 的形狀必須是可廣播的。

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

例子：

>>> t = torch.randn(1, 3)
>>> t1 = torch.randn(3, 1)
>>> t2 = torch.randn(1, 3)
>>> torch.addcdiv(t, t1, t2, value=0.1)
tensor([[-0.2312, -3.6496,  0.1312],
        [-1.0428,  3.4292, -0.1030],
        [-0.5369, -0.9829,  0.0430]])