1 引言
在论文中可能有必要讲述清楚网络的结构，可以用代码实现漂亮的绘制结构。如图所示。
2 步骤
安装两个包，本人是LInux系统，用以下命令，如果是其他系统，参考网址
sudo apt install graphviz
pip install pydot
3 实现
import tensorflow as tf
from tensorflow.keras.layers import *
from tensorflow.keras.models import Sequential, Model
model = tf.keras.Sequential()
model.add(Conv1D(8,5,strides=2,padding='same',input_shape=(4,4,8)))
model.add(LeakyReLU())
model.add(Dropout(0.3))
model.add(Conv1D(16,5, strides=2, padding='same'))
model.add(LeakyReLU())
model.add(Dropout(0.3))
model.add(Flatten())
model.add(Dense(1,activation='softmax'))
tf.keras.utils.plot_model(model, to_file='model.png', show_shapes=True)
