>>925
numpy使えるなら普通にreshapeすればよくない?

array = np.array(np.arange(320*320)).reshape([1,1,320,320])
array.shape
#=> (1, 1, 320, 320)
array2 = array.reshape([320,320,1])
array2.shape
#=> (320, 320, 1)