python矩阵的维度减少(squeeze)

用例子进行说明:
(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,)))

结果如下:
python矩阵的维度减少(squeeze)_第1张图片

(2)对最后一维进行压缩
python矩阵的维度减少(squeeze)_第2张图片

你可能感兴趣的:(python)