python如何对数组进行降维

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)

python数组快速降维

python数组多维降维、多维降为一维，快速简便

代码

X=[[[1,2,3],[4,5,6],[7,8,9]]]
X=[x for y in X for x in y]

运行结果

numpy的升维和降维

numpy的升维

a = np.array([1,2])
b = np.expand_dims(a, axis=1)

numpy的降维

a = np.array([[1],[2]])
b = np.squeeze(a)

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。