PyTorch How to get the shape of a Tensor as a list of int

Question

In numpy  V shape gives a tuple of ints of dimensions of V   In tensorflow V get shape   as list   gives a list of integers of the dimensions of V   In pytorch  V size   gives a size object  but how do I convert it to ints

User · Answer

Previous answers got you list of torch Size Here is how to get list of ints listofints    int x  for x in tensor shape

User · Answer

If you re a fan of NumPyish syntax  then there s 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   Note that tensor shape is an alias to tensor size    though tensor shape is an attribute of the tensor in question whereas tensor size   is a function

User · Answer

For PyTorch v1 0 and possibly above    gt  gt  gt  import torch  gt  gt  gt  var   torch tensor   1 0    0 1       Using  size function  returns a torch Size object   gt  gt  gt  var size   torch Size  2  2    gt  gt  gt  type var size     lt class  torch Size  gt     Similarly  using  shape  gt  gt  gt  var shape torch Size  2  2    gt  gt  gt  type var shape   lt class  torch Size  gt    You can cast any torch Size object to a native Python list    gt  gt  gt  list var size     2  2   gt  gt  gt  type list var size      lt class  list  gt      In PyTorch v0 3 and 0 4   Simply list var size     e g     gt  gt  gt  import torch  gt  gt  gt  from torch autograd import Variable  gt  gt  gt  from torch import IntTensor  gt  gt  gt  var   Variable IntTensor   1 0   0 1       gt  gt  gt  var Variable containing   1  0  0  1  torch IntTensor of size 2x2    gt  gt  gt  var size   torch Size  2  2     gt  gt  gt  list var size     2  2

[python] PyTorch: How to get the shape of a Tensor as a list of int

Examples related to python

Examples related to pytorch

Examples related to tensor