python - 如何将 4 维输入输入 LSTM？

Question

我有一个这种形状的序列输入:(6000, 64, 100, 50)

6000 只是样本序列的数量。每个序列的长度都是 64。

我计划使用 Keras 将此输入拟合到 LSTM 中。

我这样设置我的输入:

input = Input(shape=(64, 100, 50))

这给了我一个输入形状 (?, 64, 100, 50)

但是，当我像这样将 input 放入我的 LSTM 时:

x = LSTM(256, return_sequences=True)(input)

我收到这个错误:

Input 0 is incompatible with layer lstm_37: expected ndim=3, found ndim=4

如果我的输入形状类似于 (?, 64, 100)，这会起作用，但当我有第 4 维时就不行了。

这是否意味着 LSTM 只能接受 3 维的输入？如何使用 Keras 将 4 维或更高维的输入输入 LSTM？

Answer 1

你也可以考虑 TimeDistributed(LSTM(...))

inp = Input(shape=(64, 100, 50))
x = TimeDistributed(LSTM(256, return_sequences=True))(inp)

model = Model(inp, x)
model.compile('adam', 'mse')
model.summary()