KerasのLSTMの入力形式

KerasでLSTMを構築して、学習させようとしました。
train_X(X)は、(6461, 158)です。train_Y(Y)は、(6461, 1)です。
しかし、model.fit()で学習させるとエラーがはかれます。
個人的には、KerasのLSTMやRNNには、特殊な入力形式があるんだと思っているのですが、
このエラーを消すには、どうすれば良いのか教えて頂けると幸いです。

python
1# モデルの構築
2model = Sequential()
3model.add(LSTM(4, input_shape=(XX.shape[1], look_back)))
4model.add(Dense(1))
5model.compile(loss='mean_squared_error', optimizer='adam')
6model.fit(X, Y, epochs=1000, batch_size=1, verbose=2)

error
1--------------------------------------------------------------------------
2ValueError                                Traceback (most recent call last)
3<ipython-input-11-c14bfa7e740f> in <module>()
4      6 model.add(Dense(1))
5      7 model.compile(loss='mean_squared_error', optimizer='adam')
6----> 8 model.fit(X, Y, epochs=1000, batch_size=1, verbose=2)
7
82 frames
9/usr/local/lib/python3.6/dist-packages/keras/engine/training_utils.py in standardize_input_data(data, names, shapes, check_batch_axis, exception_prefix)
10    126                         ': expected ' + names[i] + ' to have ' +
11    127                         str(len(shape)) + ' dimensions, but got array '
12--> 128                         'with shape ' + str(data_shape))
13    129                 if not check_batch_axis:
14    130                     data_shape = data_shape[1:]
15
16ValueError: Error when checking input: expected lstm_2_input to have 3 dimensions, but got array with shape (6461, 158)