Pytorch中torch.numel()，torch.shape，torch.size()和torch.reshape()函数解析

一. torch.numel()函数解析

1. 官网链接

torch.numel()，如下图所示：

2. torch.numel()函数解析

torch.numel(input) → int

返回输入张量中元素的总数。

3. 代码举例

a1 = torch.randn(1, 2, 3, 4, 5)
b = torch.numel(a1)#输入元素总数为1x2x3x4x5=120
a2 = torch.zeros(4,4)
c = torch.numel(a2)#输入元素总数为4x4=16
a3 = torch.randn(size=(5,4))
d = torch.numel(a3)#输入元素总数为5x4=20
a4 = torch.randn(6)
e = torch.numel(a4)#输入元素总数为6
a1.shape,b,a2.shape,c,a3.shape,d,a4.shape,e

输出结果如下：
(torch.Size([1, 2, 3, 4, 5]),
 120,
 torch.Size([4, 4]),
 16,
 torch.Size(