keras中可以使用

from keras.utils import plot_model

plot_model(model,to_file='model_auth.png',show_shapes=True)

#show_shapes=True可以把输入输出的shape一起打印

 

注意，最好是给每个层命名，命名好之后打印出来的才会带名字。程序运行的时候也有一定的指示作用。