numpy提供了numpy.squeeze(a, axis=None)函数，从数组的形状中删除单维条目。
其中a表示输入的数组。axis用于指定需要删除的维度，但是指定的维度必须为单维度，否则将会报错。
axis的取值可为None 或 int 或 tuple of ints，若axis为空时则删除所有单维度的条目。
import numpy as np
a = np.array([[[0], [1], [2]]])
b = np.squeeze(a)
c = np.squeeze(a, 0)
d = np.squeeze(a, 2)
print('a', a)
print('a.shape', a.shape)
print('b', b)
print('b.shape', b.shape)
print('c', c)
print('c.shape', c.shape)
print('d', d)
print('d.shape', d.shape)
