torch.sub
-
torch.sub(input, other, *, alpha=1, out=None) → Tensor
-
Subtracts
other
, scaled byalpha
, frominput
.Supports broadcasting to a common shape, type promotion, and integer, float, and complex inputs.
- Parameters
- Keyword Arguments
-
-
alpha (Scalar) – the scalar multiplier for
other
- out (Tensor, optional) – the output tensor.
-
alpha (Scalar) – the scalar multiplier for
Example:
>>> a = torch.tensor((1, 2)) >>> b = torch.tensor((0, 1)) >>> torch.sub(a, b, alpha=2) tensor([1, 0])
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.sub.html