np.squeeze()

np.squeeze() 是 NumPy 库中的一个函数，用于从数组中删除单维度的条目。它返回一个在输入数组中删除了尺寸为 1 的维度的新数组。

下面是使用 np.squeeze() 的示例代码：

import numpy as np

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

squeezed_arr = np.squeeze(arr)
print(squeezed_arr.shape)  # (3,)
print(squeezed_arr)

输出：

(1, 3, 1)
(3,)
[1 2 3]

在上述示例中，我们创建了一个形状为 (1, 3, 1) 的输入数组 arr。然后，我们使用 np.squeeze() 函数将其变形为 (3,) 的数组 squeezed_arr。通过打印 squeezed_arr，我们可以看到被压缩后的数组中不再有尺寸为 1 的维度。