making a picture of a neural network in tensorflow
This should do the trick:
tf.keras.utils.plot_model(
        model, to_file='model.png', show_shapes=False, show_layer_names=True,
        rankdir='TB', expand_nested=False, dpi=96
    )

This will generate an image like:

Example: https://www.tensorflow.org/guide/keras/functional