python数组进行降维

在深度学习训练过程中，我们有时候想要输出图片看看图片长什么样，但是训练时的图片格式一般都会多出一个批次的维度，如[1,400,400,3]，这时我们需要把第1个维度去掉才能展示图片：

（1）如果是array的格式，可以直接：img = img.squeeze()

（2）如果是tensor格式图片，可以用img = tf.squeeze(img, [0])

（3）tensor格式图片想增加维度：img = tf.expand_dims(img, 0)