如何更改神经网络图像数据形状的维度顺序？

Question

我有一个灰度图像数据集，我想将其输入神经网络。

x_train_grey.shape 是 (32, 32, 73257) 所以我知道它是 (dimension_x, dimension_y, batch_size)。因为图像是灰度的，所以只有一个“深度”维度。

但是，要将这些数据提供给神经网络，它需要具有以下形状：(batch_size, dimension_x, dimension_y)。以batch_szie 开头。 如何将其重塑为这种格式，以便 batch_szie 出现在 x、y 图像尺寸之前？

一旦完成，我希望能够将其传递到神经网络（第一层是Flatten()），如下所示： Flatten(input_shape=(32, 32, 1)),.

干杯！

Answer 1

解决了！通过将正确的形状传递给np.reshape()。

在进入深度学习之前，我真的应该更好地了解numpy。