参考
https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int

<https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int>
>>> 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]