LSTM的初始状态

Question

我想检查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)

get_initial_states所需的输入形状是什么？ 这是检查LSTM的正确方法