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.
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]
<!–
–>
Please Login to comment…