數組改變維度

首先創建一個數組,維度是[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數組的排列方式其實沒有變，

>>阅读原文<<