原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.squeeze.html
校对:(虚位以待)
numpy.
squeeze
(a, axis=None)[source]从数组的形状中删除单维条目。
参数: | a:array_like
axis:无或int或tuple ints,可选
|
---|---|
返回: | 挤压:ndarray
|
例子
>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
>>> np.squeeze(x).shape
(3,)
>>> np.squeeze(x, axis=(2,)).shape
(1, 3)