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

PyTorch torch.trunc()移除數字的小數部分後，方法會返回一個新的張量，其中包含input /元素的截斷整數值。

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

參數

input:這是輸入張量。
out:輸出張量。

返回：它返回張量。

讓我們借助幾個示例來了解這個概念：
範例1：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.randn(6) 
print(a) 
  
# Applying the trunc function and  
# storing the result in 'out' 
out = torch.trunc(a) 
print(out)

輸出：

 1.1257
 0.4493
-0.7309
 1.5523
-0.2877
 0.1155
[torch.FloatTensor of size 6]
 1
 0
-0
 1
-0
 0
[torch.FloatTensor of size 6]

範例2：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatTensor([1.5, 3.9, -6.9, 3.678]) 
print(a) 
  
# Applying the trunc function and  
# storing the result in 'out' 
out = torch.trunc(a) 
print(out)

輸出：

 1.5000
 3.9000
-6.9000
 3.6780
[torch.FloatTensor of size 4]
 1
 3
-6
 3
[torch.FloatTensor of size 4]

相關用法

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