如何在Tensorflow中获得LSTM的密集层输出？

import tensorflow as tf
from tensorflow.keras.models import Model
from tensorflow.keras.layers import LSTM, Dense

input_shape = (batch_size, time_steps, input_dim)
lstm_units = 64
dense_units = 32

input_layer = tf.keras.Input(shape=input_shape)
lstm_layer = LSTM(lstm_units, return_sequences=True)(input_layer)
dense_layer = Dense(dense_units)(lstm_layer)

model = Model(inputs=input_layer, outputs=dense_layer)

model.compile(optimizer='adam', loss='mse')
model.fit(x_train, y_train, epochs=10, batch_size=batch_size)

lstm_output_model = Model(inputs=model.input, outputs=model.layers[1].output)
lstm_output = lstm_output_model.predict(x_test)