创建一个shape为【13224】的5维tensor数组值为0到48

下面是创建这样一个tensor的代码：

import torch

arr = torch.arange(0, 48).reshape(1, 3, 2, 2, 4)
print(arr.shape)
print(arr)

输出：

torch.Size([1, 3, 2, 2, 4])
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],
           [44, 45, 46, 47]]]]])