有一个nd数组A，形状为[100,255,255,3]，对应100 255*255图像。我想迭代这个多维数组，每次迭代我都会得到一张图像。这就是我所做的，A1 = A[i,:,:,:] 生成的 A1 具有形状 [255,255,3]。但是，我想强制它具有形状 [1,255,255,3]。我该怎么做？
最佳答案
np.reshape(A1, (1, 255, 255, 3))
应该可以解决问题
关于python - 为多维数组添加虚拟维度，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/40231128/