从打印输出的张量确定维度

tensor(  [[[[ 0.4024,  0.8702],
          [-0.0625, -0.3826]],

         [[-1.1489, -0.5550],
          [ 0.0176,  1.0890]]]])
          
 tensor( [[[ 0.4024,  0.8702, -0.0625, -0.3826],
         [-1.1489, -0.5550,  0.0176,  1.0890]]])

这两个tensor的维度是怎么样的?
从最外层的大括号开始,最外层的内容有一项

         [[[ 0.4024,  0.8702],
          [-0.0625, -0.3826]],

         [[-1.1489, -0.5550],
          [ 0.0176,  1.0890]]]

第二层有两项

         [[ 0.4024,  0.8702],
          [-0.0625, -0.3826]],

         [[-1.1489, -0.5550],
          [ 0.0176,  1.0890]]

第三层有两项

          [ 0.4024,  0.8702],
          [-0.0625, -0.3826]

第四层有两项

           0.4024,  0.8702

因此这个张量维度为1* 2* 2* 2

猜你喜欢

转载自blog.csdn.net/m0_38133212/article/details/88780358
今日推荐