pytorch / 2 / generated / torch.sub.html

torch.sub

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

Subtracts other, scaled by alpha, from input.

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

Supports broadcasting to a common shape, type promotion, and integer, float, and complex inputs.

Parameters
  • input (Tensor) – the input tensor.
  • other (Tensor or Number) – the tensor or number to subtract from input.
Keyword Arguments
  • alpha (Number) – the multiplier for other.
  • out (Tensor, optional) – the output tensor.

Example:

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

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.sub.html