np.reshape()

np.reshape(5, 6)
生成二维数组，内涵5个一维数组，每个数组6个元素。
np.reshape(4, 3, 2)
生成三维数组， 内涵4个二维数组，每个二维数组3行2列。

import numpy as np
import tensorflow as tf

a = np.arange(30).reshape(5, 6)
b = np.arange(24).reshape(4, 3, 2)

print(a)
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]]

[[[ 0  1]
  [ 2  3]
  [ 4  5]]

 [[ 6  7]
  [ 8  9]
  [10 11]]

 [[12 13]
  [14 15]
  [16 17]]

 [[18 19]
  [20 21]
  [22 23]]]