如何重塑4D Numpy数组以训练模型

Question

我正在尝试训练我已经编译的CNN和LSTM模型。但是我不断收到错误

  

ValueError：检查输入时出错：预期   time_distributed_151_input具有5个维度，但具有   形状（4732，32，32，3）

我的模型摘要看起来像这样

图层（类型）输出形状参数＃

time_distributed_151（TimeDi（None，3，32，13，13）4736

---

time_distributed_152（TimeDi（None，3，32，11，11）9248

---

time_distributed_153（TimeDi（None，3，32，5，5）0

---

我似乎无法将输入形状扩展到期望的5维数组。在网上查看了各种资料之后，我对能够重塑我的Numpy数组没有好运。

任何帮助都会很棒

Answer 1

你可以使用这段代码：

x = np.array([[4732, 32, 32, 3]])
x.reshape(x.shape[0],-1)

我希望它有效。