用例子进行说明:
(1)对0维进行压缩
x = np.array([[[0], [1], [2]]])
print(x)
x.shape
(1, 3, 1)
np.squeeze(x).shape
np.squeeze(x, axis=(,)).shape
print(np.squeeze(x, axis=(0,)))
结果如下:
(2)对最后一维进行压缩
用例子进行说明:
(1)对0维进行压缩
x = np.array([[[0], [1], [2]]])
print(x)
x.shape
(1, 3, 1)
np.squeeze(x).shape
np.squeeze(x, axis=(,)).shape
print(np.squeeze(x, axis=(0,)))
结果如下:
(2)对最后一维进行压缩