torch / torch
torch.add¶
-
torch.
add
(input, other, *, out=None)¶ Adds the scalar
other
to each element of the inputinput
and returns a new resulting tensor.If
input
is of type FloatTensor or DoubleTensor,other
must 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
other
is multiplied by the scalaralpha
and added to each element of the tensorinput
. The resulting tensor is returned.The shapes of
input
andother
must be broadcastable.If
other
is of type FloatTensor or DoubleTensor,alpha
must 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.
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]])
此页内容是否对您有帮助