[人工智能]np.squeeze 的用法

我在可视化我的数据集的时候，我发现有些不同
比如我同样的代码可视化CIFAR10数据集是没有问题的
但是我去可视化MINST这种没有RGB数据集的时候，我发现用Pytorch的loader不能直接可视化

我发现，这里面可视化我们需要np.squeeze

squeeze 函数：从数组的形状中删除单维度条目，即把shape中为1的维度去掉
所以如果我们的图片为28x28x1，我们可以用这个函数直接可视化图片

>>> a = e.reshape(1,1,10)
>>> a
array([[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]]])
>>> np.squeeze(a)
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

import numpy as np
x = np.array([[[0], [1], [2]]])
print(x.shape)

d = np.squeeze(x)   # 从数组的形状中删除单维条目，即把shape中为1的维度去掉
print(d.shape)

(1,3,1)
(3,)