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