Python - PyTorch add() method

PyTorch torch.add() method adds a constant value to each element of the input tensor and returns a new modified tensor.

Syntax: torch.add(inp, c, out=None) Arguments

• 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.

Return: It returns a Tensor.

Let's see this concept with the help of few examples: Example 1: Python3

# 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)

Output:

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: Python3

# 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) 

Output:

 1
  3
  8
  4
 10
[torch.FloatTensor of size 5]
  6
  8
 13
  9
 15
[torch.FloatTensor of size 5]