x = np.array([[[0], [1], [2]]]) x.shape (1, 3, 1) np.squeeze(x).shape (3,) np.squeeze(x, axis=0).shape (3, 1) np.squeeze(x, axis=1).shape