我就废话不多说了,直接上代码吧!
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
>>> 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 ] |
参考https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int
以上这篇pytorch 获取tensor维度信息示例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/u010454261/article/details/83152526