我正在使用Python库Keras运行一些LSTM时间预测。 源代码确实是高级的,我正在尝试研究库以更好地理解中间变量,尤其是那些变量的尺寸,例如输入,权重等。
model.add(LSTM(7, input_shape=(time_number, feature_number)))
这一行几乎建立了模型。我设法找到了真正发生魔术的代码部分(Kera Github LSTMcell source code)。我想做一些事情,例如在每个步骤中添加一条打印语句,以打印出每个变量的尺寸,输入尺寸,x_i,x_u,o,i等。
我可以进入Keras的安装位置并在那里调整源代码,但是我想知道是否有一种优雅的方式使用诸如inspect之类的库来保留这些中间变量以进行检查。
我发现有this帖子,但是它只给出了内核和recurrent_kernel大小的尺寸,我希望检查其他许多变量的尺寸。