import numpy as np

x = np.array([[[0], [1], [2]]])

print(x.shape)

d = np.squeeze(x)   # 从数组的形状中删除单维条目，即把shape中为1的维度去掉

print(d.shape)