np.squeeze() 是 NumPy 库中的一个函数，用于从数组中删除单维度的条目。它返回一个在输入数组中删除了尺寸为 1 的维度的新数组。
下面是使用 np.squeeze() 的示例代码：
点击查看代码
import numpy as np
arr=np.array([[[1,2,3],[2,3,4]]])
print(arr.shape)
sqeezed_arr=np.squeeze(arr)
print(sqeezed_arr.shape)
sqeezed_arr=np.squeeze(sqeezed_arr)
print(sqeezed_arr.shape)
点击查看代码
(1, 2, 3)
(2, 3)
(2, 3)