np.squeeze(a, axis=None)将矩阵维度进行移除
如:
>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
>>> np.squeeze(x).shape
(3,)
>>> np.squeeze(x, axis=0).shape
(3, 1)
这里当axis=None,即默认值时,np.squeeze将矩阵直接变为一维数组
如果你指定了轴,那么他将移除指定轴的维度
并且不能移除元素个数大于1的轴
>>> np.squeeze(x, axis=1).shape
Traceback (most recent call last):
...
ValueError: cannot select an axis to squeeze out which has size not equal to one