【发布时间】:2019-07-12 09:33:31
【问题描述】:
当使用 keras 功能 model.summary() 时,它向我展示了我的模型的张量形状,非常棒!
不幸的是,当使用编码器 LSTM 时,使用带有属性 return_states=True 的 keras.layers.LSTM 构造函数调用,摘要不会以其完整形式显示。它看起来像这样:
Layer (type) Output Shape Param # Connected to
==================================================================================================
input (InputLayer) (None, 34, 30) 0
__________________________________________________________________________________________________
encoder (LSTM) [(None, 34, 30), (No 7320 input[0][0]
__________________________________________________________________________________________________
lambda_8 (Lambda) (None, 34, 15) 0 encoder[0][0]
__________________________________________________________________________________________________
decoder (LSTM) (None, 34, 30) 5520 lambda_8[0][0]
encoder[0][1]
encoder[0][2]
==================================================================================================
Total params: 12,840
Trainable params: 12,840
Non-trainable params: 0
__________________________________________________________________________________________________
如您所见,编码器的输出形状被截断,只有三个形状中的第一个可见。有没有办法显示它,也许是修复甚至解决方法? :)
【问题讨论】:
标签: keras deep-learning state lstm summary