numpy.squeeze

numpy.squeeze(a,axis=None)
从数组形状中移除一维条目

参数：
a – 数组形状np.array()
axis – 可以是一个整数，可以是一个元组或者None。默认为None。axis用于指定需要删除的维度，但是指定的维度必须为单维度，否则将会报错.

返回：压缩后的ndarray()
减少了维度的数据子集

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

#axis=None
>>> a = np.squeeze(x)
>>> a
array([0, 1, 2])
>>> a.shape	#不指定轴，将全部一维的条目压缩
(3,)

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