torch.add
-
torch.add(input, other, *, out=None) -
Adds the scalar
otherto each element of the inputinputand returns a new resulting tensor.If
inputis of type FloatTensor or DoubleTensor,othermust be a real number, otherwise it should be an integer.- Parameters
-
- input (Tensor) – the input tensor.
-
value (Number) – the number to be added to each element of
input
- Keyword Arguments
-
out (Tensor, optional) – the output tensor.
Example:
>>> a = torch.randn(4) >>> a tensor([ 0.0202, 1.0985, 1.3506, -0.6056]) >>> torch.add(a, 20) tensor([ 20.0202, 21.0985, 21.3506, 19.3944])
-
torch.add(input, other, *, alpha=1, out=None)
Each element of the tensor
otheris multiplied by the scalaralphaand added to each element of the tensorinput. The resulting tensor is returned.The shapes of
inputandothermust be broadcastable.If
otheris of type FloatTensor or DoubleTensor,alphamust be a real number, otherwise it should be an integer.- Parameters
- Keyword Arguments
-
-
alpha (Number) – the scalar multiplier for
other - out (Tensor, optional) – the output tensor.
-
alpha (Number) – the scalar multiplier for
Example:
>>> a = torch.randn(4) >>> a tensor([-0.9732, -0.3497, 0.6245, 0.4022]) >>> b = torch.randn(4, 1) >>> b tensor([[ 0.3743], [-1.7724], [-0.5811], [-0.8017]]) >>> torch.add(a, b, alpha=10) tensor([[ 2.7695, 3.3930, 4.3672, 4.1450], [-18.6971, -18.0736, -17.0994, -17.3216], [ -6.7845, -6.1610, -5.1868, -5.4090], [ -8.9902, -8.3667, -7.3925, -7.6147]])
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.add.html