torch.sub

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

Вычитает other, умноженное на alpha, из input.

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

Поддерживает broadcasting до общей формы, приведение типов, а также целочисленные, вещественные и комплексные входные данные.

Параметры:

• input (Tensor) – входной тензор.
• other (Tensor или Number) – тензор или число, которое нужно вычесть из input.

Именованные аргументы:

• alpha (Number) – множитель для other.
• out (Tensor, optional) – выходной тензор.

Пример:

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