keras中LSTM的输入问题

当LSTM在Sequential第一层时，需要设置它的输入形状，为[time_stamp,input_dim],举例来说，有一段数据如下：

X = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
X = X.reshape(2,5,1)
timesteps=5
data_dim=1
model = keras.Sequential()
model.add(keras.layers.LSTM(32, return_sequences=True, input_shape=(timesteps, data_dim)))

这样就表示一共有两段数据，每段数据长度为5，而每段数据只有一个特征，如果每段数据有两个特征就修改data_dim为2