pytorch 获取tensor维度信息

参考
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]

• « 上一篇：一份数据科学家必备的技能清单（附资源）
• » 下一篇：安卓面试中高级安卓开发工程师总结之——如何写一份让HR主动邀请你面试的简历