reshape.py

A = np.arange(8 * 4).reshape(8, 4)
print(A)
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]
#  [12 13 14 15]
#  [16 17 18 19]
#  [20 21 22 23]
#  [24 25 26 27]
#  [28 29 30 31]]
B = np.arange(4 * 8).reshape(4, 8)
print(B)
# [[ 0  1  2  3  4  5  6  7]
#  [ 8  9 10 11 12 13 14 15]
#  [16 17 18 19 20 21 22 23]
#  [24 25 26 27 28 29 30 31]]
A = A.reshape(2, 2, 4, 2)
print(A)
# [[[[ 0  1]
#    [ 2  3]
#    [ 4  5]
#    [ 6  7]]
#   [[ 8  9]
#    [10 11]
#    [12 13]
#    [14 15]]]
#  [[[16 17]
#    [18 19]
#    [20 21]
#    [22 23]]
#   [[24 25]
#    [26 27]
#    [28 29]
#    [30 31]]]]
B = B.reshape(2, 2, 2, 4)
print(B)
# [[[[ 0  1  2  3]
#    [ 4  5  6  7]]
#   [[ 8  9 10 11]
#    [12 13 14 15]]]
#  [[[16 17 18 19]
#    [20 21 22 23]]
#   [[24 25 26 27]
#    [28 29 30 31]]]]