Python LSTM的初始状态

我想检查LSTM层的内部状态。 在向网络提供了一个新的示例之后，我特别想看看预测时的状态

我理解这可以通过以下方式实现：

from keras import backend as K
# load pre-trained model somewhere
# select a LSTM layer
for layer in model.layers:
    if 'LSTM' in str(layer):
       break
# get inputs somewhere

val = np.random.random((...))
x = K.variable(value=val)

initial_state = layer.get_initial_states(???)
[desired_states]=layer.step(inputs=x,states=initial_state)

获取初始状态所需的输入形状是什么？ 这是检查LSTM的正确方法吗