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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。