python - 如何重塑 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 数组。

任何帮助都会很棒

标签： pythonarraysnumpytensorflowlstm

Answer 1

您可以使用这段代码：

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

我希望它有效。