numpy.squeeze

`numpy.squeeze` 是 NumPy 库中的一个函数，用于从数组的维度中去除长度为1的维度。这个函数的作用是对输入数组进行操作，返回一个将长度为1的维度去除的新数组。

具体来说，`numpy.squeeze` 的函数签名如下：

numpy.squeeze(a, axis=None)

其中参数 `a` 是输入的数组，而参数 `axis` 是可选的整数或整数元组，表示要删除的指定轴。如果 `axis` 是整数，则表示删除指定索引的轴；如果 `axis` 是元组，则表示删除指定索引的多个轴。

下面是一个示例：

import numpy as np

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

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

在这个示例中，输入数组 `a` 的形状是 `(1, 3, 1)`，其中有一个长度为1的维度。使用 `numpy.squeeze` 函数将这个长度为1的维度去除后，得到的新数组 `b` 的形状是 `(3,)`。