Open In App

Python – PyTorch fmod() method

Improve
Improve
Like Article
Like
Save
Share
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:




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




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

Last Updated : 26 May, 2020
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads