Open In App

Python - PyTorch add() method

Last Updated : 26 May, 2020
Comments
Improve
Suggest changes
Like Article
Like
Report
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]

Next Article
Article Tags :
Practice Tags :

Similar Reads