Python - PyTorch fmod() method

Last Updated : 26 May, 2020
Comments
Improve
Suggest changes
Like Article
Like
Report
PyTorch torch.fmod() method gives the element-wise remainder of division by divisor. The divisor may be a number or a Tensor.
Syntax: torch.fmod(input, div, out=None) Arguments
  • input: This is input tensor.
  • div: This may be a number or a tensor.
  • out: 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 n
a = torch.FloatTensor([5, 6, 7, 4])
print(a)

# Applying the fmod function and 
# storing the result in 'out'
out = torch.fmod(a, 3)
print(out)
Output: 
5
 6
 7
 4
[torch.FloatTensor of size 4]
 2
 0
 1
 1
[torch.FloatTensor of size 4]
Example 2: Python3 
# Importing the PyTorch library 
import torch 
  
# A constant tensor of size n
a = torch.FloatTensor([5, 6, 7, 4])
b = torch.FloatTensor([2, 3, 4, 1])
print(a, b)

# Applying the fmod function and 
# storing the result in 'out'
out = torch.fmod(a, b)
print(out)
Output: 
5
 6
 7
 4
[torch.FloatTensor of size 4]
 2
 3
 4
 1
[torch.FloatTensor of size 4]

 1
 0
 3
 0
[torch.FloatTensor of size 4]

Next Article
Python - PyTorch add() method

P
PranchalKatiyar
Improve
Article Tags :
Practice Tags :

Similar Reads