np.squeeze()

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)

并且验证了如果没有尺寸为1的维度也不会报错