pytorch中张量的阶数理解

pytorch中张量的阶数理解

推荐打开2个页面，对比原四阶张量理解各阶的对应关系。

• 创建一个四阶张量：

import torch
x = torch.linspace(0,71,72).view(2,3,3,4)
print(x)

• 输出：

tensor([[[[ 0.,  1.,  2.,  3.],
          [ 4.,  5.,  6.,  7.],
          [ 8.,  9., 10., 11.]],
          
          [[12., 13., 14., 15.],
          [16., 17., 18., 19.],
          [20., 21., 22., 23.]],

         [[24., 25., 26., 27.],
          [28., 29., 30., 31.],
          [32., 33., 34., 35.]]],


        [[[36., 37., 38., 39.],
          [40., 41., 42., 43.