Python 3.x 重塑和添加尺寸

我有一个y_序列（4D）矩阵，其内容是（332242241），其中（n_图像，n_像素，n_像素，n_x，n_波段），需要使用展平将其转换为2D向量

现在我需要做相反的过程，把2D向量转换回具有相同维数的4D矩阵

我使用了np.reforme（）和np.expand_dims（）来恢复形状（224224，224，1），但我需要将“n_图像”添加到（n_图像，224224，224，1） 最好的方法是什么

train_x=np.整形（y_测试[1]，（224224，224））
打印（序列x.形状）#此结果为（224224，224）
列=np.展开尺寸（列x，-1）
打印（train.shape）#此结果为（224224，224，1）
pred=指数检验
打印（pred.shape）##此结果为（33，）

将-1作为新形状的一维尺寸，numpy将计算出正确的尺寸您可以在重塑中包含展开（224224,1）在重塑

y\u测试[1]时不能包含33。它不够大。