Python - PyTorch add() method
Last Updated :
26 May, 2020
Improve
PyTorch
Python3
Output:
Python3
Output:
torch.add()
method adds a constant value to each element of the input tensor and returns a new modified tensor.
Syntax:Let's see this concept with the help of few examples: Example 1:torch.add(inp, c, out=None)
ArgumentsReturn: It returns a Tensor.
- inp: This is input tensor.
- c: The value that is to be added to every element of tensor.
- out: This is optional parameter and it is the output tensor.
# Importing the PyTorch library
import torch
# A constant tensor of size 6
a = torch.randn(6)
print(a)
# Applying the add function and
# storing the result in 'b'
b = torch.add(a, 5)
print(b)
0.2403 1.3826 -0.1763 -1.5177 -0.0555 1.4558 [torch.FloatTensor of size 6] 5.2403 6.3826 4.8237 3.4823 4.9445 6.4558 [torch.FloatTensor of size 6]Example 2:
# Importing the PyTorch library
import torch
# A constant tensor of size 6
a = torch.FloatTensor([1, 3, 8, 4, 10])
print(a)
# Applying the add function and
# storing the result in 'b'
b = torch.add(a, 5)
print(b)
1 3 8 4 10 [torch.FloatTensor of size 5] 6 8 13 9 15 [torch.FloatTensor of size 5]