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

PyTorch torch.ceil()方法返回具有輸入元素的ceil值的新張量，該張量是大於或等於每個元素的最小整數。

用法： torch.ceil(inp, out=None)

參數

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

返回：它返回張量。

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

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatTensor([1.3, -5.6, 7.9, 10.1]) 
print(a) 
  
# Applying the ceil function and  
# storing the result in 'out' 
out = torch.ceil(a) 
print(out)

輸出：

  1.3000
 -5.6000
  7.9000
 10.1000
[torch.FloatTensor of size 4]
  2
 -5
  8
 11
[torch.FloatTensor of size 4]

範例2：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatTensor([1.00001, -5.699, 7.100001, 10.0001]) 
print(a) 
  
# Applying the ceil function and  
# storing the result in 'out' 
out = torch.ceil(a) 
print(out)

輸出：

 1.0000
 -5.6990
  7.1000
 10.0001
[torch.FloatTensor of size 4]
  2
 -5
  8
 11
[torch.FloatTensor of size 4]

相關用法

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