数组改变维度

首先创建一个数组,维度是[2,3,4,5]

import torch
import numpy as np
a=np.array(range(120)).reshape(2,3,4,5)

 可以看到数字是依次排列的，从最后一个维度，也就是站从[5]这个维度慢慢开始填充，填充了5个元素之后，开始扩散到[4]这个维度。

b=a.reshape(5,4,3,2)

可以看到b和a数组的排列方式其实没有变，变得只是括号，也就是[]的位置