Skip to content
Related Articles

Related Articles

Improve Article
Python – PyTorch add() method
  • Last Updated : 26 May, 2020

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:






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




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

 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course




My Personal Notes arrow_drop_up
Recommended Articles
Page :