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…