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$$

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

Параметры

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

Ключевые аргументы

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

Пример:

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