在深度学习训练过程中，我们有时候想要输出图片看看图片长什么样，但是训练时的图片格式一般都会多出一个批次的维度，如[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)
在深度学习训练中，为了查看图片，通常需要处理批次维度。对于array格式的图片，可以使用img=img.squeeze()去除第1个维度；如果是tensor，可以使用tf.squeeze(img,[0])来移除。若需增加维度，可以利用tf.expand_dims(img,0)。这些操作对于理解和调试模型至关重要。
2500
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
