PyTorch：如何得到张量的形状的INT

Question

在numpy的，V.shape列表给出了五

的维整数的元组

在tensorflow V.get_shape().as_list()给出了V的尺寸的整数列表

在pytorch中，V.size()给出了一个大小对象，但是如何将它转换为整数？

Answer 1

简单list(var.size())，例如：

>>> import torch 
>>> from torch.autograd import Variable 
>>> from torch import IntTensor 
>>> var = Variable(IntTensor([[1,0],[0,1]])) 

>>> var 
Variable containing: 
1 0 
0 1 
[torch.IntTensor of size 2x2] 

>>> var.size() 
torch.Size([2, 2]) 

>>> list(var.size()) 
[2, 2] 

Answer 2

如果你NumPy ISH语法的粉丝，那么就tensor.shape。

In [3]: ar = torch.rand(3, 3) 

In [4]: ar.shape 
Out[4]: torch.Size([3, 3]) 

# method-1 
In [7]: list(ar.shape) 
Out[7]: [3, 3] 

# method-2 
In [8]: [*ar.shape] 
Out[8]: [3, 3] 

# method-3 
In [9]: [*ar.size()] 
Out[9]: [3, 3] 

P.S.：请注意tensor.shape是tensor.size()的别名，尽管tensor.shape是所讨论的张量的属性，而tensor.size()是函数。只有在启用了GPU的机器上才能使用。