编辑:抱歉,我没有意识到这是 3 维结果,而不是 4 维结果。要获得 3D 形状,您必须再次重塑。还有额外的重塑will复制数据。
你不能,你还需要转置:
In [1]: a = np.arange(8)[:,None].repeat(8,axis=1)
In [2]: a
Out[2]:
array([[0, 0, 0, 0, 0, 0, 0, 0],
[1, 1, 1, 1, 1, 1, 1, 1],
[2, 2, 2, 2, 2, 2, 2, 2],
[3, 3, 3, 3, 3, 3, 3, 3],
[4, 4, 4, 4, 4, 4, 4, 4],
[5, 5, 5, 5, 5, 5, 5, 5],
[6, 6, 6, 6, 6, 6, 6, 6],
[7, 7, 7, 7, 7, 7, 7, 7]])
In [3]: b = a.reshape(2,4,2,4)
In [4]: b
Out[4]:
array([[[[0, 0, 0, 0],
[0, 0, 0, 0]],
...
[[7, 7, 7, 7],
[7, 7, 7, 7]]]])
In [5]: b.transpose(0,2,1,3)
Out[5]:
array([[[[0, 0, 0, 0],
[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3]],
[[0, 0, 0, 0],
[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3]]],
[[[4, 4, 4, 4],
[5, 5, 5, 5],
[6, 6, 6, 6],
[7, 7, 7, 7]],
[[4, 4, 4, 4],
[5, 5, 5, 5],
[6, 6, 6, 6],
[7, 7, 7, 7]]]])