Python PyTorch fmod()用法及代碼示例

PyTorch torch.fmod()方法給出除數除以元素的餘數。除數可以是數字或張量。

用法： torch.fmod(input, div, out=None)

參數

input:這是輸入張量。
div:這可以是數字或張量。
out:輸出張量。

返回：它返回張量。

讓我們借助幾個示例來了解這個概念：
範例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)

輸出：

5
 6
 7
 4
[torch.FloatTensor of size 4]
 2
 0
 1
 1
[torch.FloatTensor of size 4]

範例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)

輸出：

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]

相關用法

注：本文由純淨天空篩選整理自PranchalKatiyar大神的英文原創作品 Python – PyTorch fmod() method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。