np.transpose()转换维度

import numpy as np

x = np.arange(24).reshape((4,3,2))
print(x.shape)

x = np.transpose(x,[1,2,0])
print(x.shape)


# 结果
(4, 3, 2)
(3, 2, 4)

原本维度 => 4，3，2

新维度 => 3，2，4

np.tranpose([a,b,c])其中a,b,c分别代表的是旧维度在新维度所处的具体位置

例如：

旧数据中第0个维度是4，经过np.transpose([a,b,0])处理

新数据中第2个维度即 旧数据的第0个维度(4)

以此类推