Python PyTorch sub用法及代碼示例

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

用法:

torch.sub(input, other, *, alpha=1, out=None) → Tensor

參數：

• input(Tensor) -輸入張量。

• other(Tensor或者數字) -從 input 中減去的張量或數字。

關鍵字參數：

• alpha(數字) -other 的乘數。

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

從 input 減去由 alpha 縮放的 other 。

\text{{out}}_i = \text{{input}}_i - \text{{alpha}} \times \text{{other}}_i

支持廣播到通用形狀、類型提升以及整數、浮點數和複雜輸入。

例子：

>>> a = torch.tensor((1, 2))
>>> b = torch.tensor((0, 1))
>>> torch.sub(a, b, alpha=2)
tensor([1, 0])