tf.keras.utils.plot_model

TensorFlow 2 version View source on GitHub

Converts a Keras model to dot format and save to a file.

model A Keras model instance
to_file File name of the plot image.
show_shapes whether to display shape information.
show_layer_names whether to display layer names.
rankdir rankdir argument passed to PyDot, a string specifying the format of the plot: 'TB' creates a vertical plot; 'LR' creates a horizontal plot.
expand_nested Whether to expand nested models into clusters.
dpi Dots per inch.

A Jupyter notebook Image object if Jupyter is installed. This enables in-line display of the model plots in notebooks.