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

PyTorch torch.numel()方法返回輸入張量中元素的總數。

用法： torch.numel(input)

參數

input:這是輸入張量。

返回：它返回輸入張量的長度。

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

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

輸出：

-0.8263  0.9807 -1.4688  0.2117 -0.8356 -0.0228
-0.8815  1.3652 -0.1892 -1.1241  0.2755  1.3006
 0.0559  0.2389  0.7944  2.6587 -2.0908  1.2973
-0.2056  0.4110  0.2163  0.3091  0.5559 -0.2468
[torch.FloatTensor of size 4x6]
24

範例2：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n 
a = torch.FloatTensor([1, 4, 6, 8]) 
print(a) 
  
# Applying the numel function and  
# storing the result in 'out' 
out = torch.numel(a) 
print(out)

輸出：

 1
 4
 6
 8
[torch.FloatTensor of size 4]
4

相關用法

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