У меня есть матрица y_train (4D), содержимое которой (33, 224, 224, 1) где (n_images, n_pixels_y, n_pixels_x, n_bands), которые необходимо преобразовать в двухмерный вектор с использованием Flatten
Теперь мне нужно выполнить обратный процесс и повернуть 2D-вектор обратно в 4D-матрицу с теми же размерами.
Я использовал np.reshape () и np.expand_dims (), чтобы получить форму (224, 224, 1) назад, но мне нужно добавить 'n_images' в (n_images, 224, 224, 1), как это лучше всего сделать?
train_x = np.reshape(y_test[1], (224, 224))
print(train_x.shape) #this result in (224, 224)
train = np.expand_dims(train_x, -1)
print(train.shape) #this result in (224, 224, 1)
pred = index_test
print(pred.shape) ##this result in (33,)