numpy squeeze函数

`numpy.squeeze`函数是用来从数组的形状中删除单维度条目的。它返回一个已移除单维度条目的新数组。

语法：

numpy.squeeze(arr, axis=None)

参数说明：

* `arr`：输入的数组。
* `axis`：整数或整数元组，用于选择单维度子数组。默认值为None，表示所有单维度条目都被删除。

返回值：已移除单维度条目的新数组。

示例：

import numpy as np

a = np.array([[[0], [1], [2]]])
print(a.shape)  # (1, 3, 1)

b = np.squeeze(a)
print(b.shape)  # (3,)

c = np.squeeze(a, axis=0)
print(c.shape)  # (3, 1)

输出：

(1, 3, 1)
(3,)
(3, 1)