Tensorflow图显示所有正在调用的计算。你将无法简化它。
作为替代方案,Keras拥有自己的逐层图形。这显示了您的网络清晰简洁的结构。你可以通过调用
from keras.utils import plot_model
plot_model(model, to_file='/some/pathname/model.png')
最后生成它,你也可以拨打model.summary()
,其生成图形的文字版本,额外的摘要。
这里是model.summary()
例如一个输出:
Layer (type) Output Shape Param # Connected to
====================================================================================================
input_1 (InputLayer) (None, 2048) 0
____________________________________________________________________________________________________
activation_1 (Activation) (None, 2048) 0
____________________________________________________________________________________________________
dense_1 (Dense) (None, 511) 1047039
____________________________________________________________________________________________________
activation_2 (Activation) (None, 511) 0
____________________________________________________________________________________________________
decoder_layer_1 (DecoderLayer) (None, 512) 0
____________________________________________________________________________________________________
ctg_output (OrLayer) (None, 201) 102912
____________________________________________________________________________________________________
att_output (OrLayer) (None, 312) 159744
====================================================================================================
Total params: 1,309,695.0
Trainable params: 1,309,695.0
Non-trainable params: 0.0
感谢。 'plot_model'不能立即生效,我必须安装'pydot_ng','graphviz'和Graphviz软件。但现在它起作用了。 – StayFoolish