如何在Keras中為LSTM模型正確指定輸入形狀

Question

假設我們有一個Numpy數組，其中包含整數。

arr = [1, 2, 3]

我們有一個Keras模型。

model.add(LSTM(128, input_shape=(arr)))

如何正確指定輸入形狀？

Answer 1

LSTM層的輸入數據必須是三維的，並且形狀為(num_samples, timesteps, num_features) 。

您為LSTM圖層指定的input_shape具有形狀(timesteps, num_features) 。 input_shape並不關心樣本數量，僅關心每個樣本的形狀。

如果我們假設timesteps=1 ，那么您想要做這樣的事情。

arr = np.array([1, 2, 3])
arr.shape  # (3,)
arr = arr.reshape(arr.shape[0], 1, 1)
arr.shape  # (3, 1, 1)

model.add(LSTM(128, input_shape=(arr.shape[1], arr.shape[2])))

在timesteps=1使用LSTM並沒有多大意義，但希望您能理解。