Is there any way to obtain the computation graph of an NN?

Hi,

I’d like to get the computation graph of a self-defined NN, is there any way in tensorboard that could help me to achieve that?

The Tensorboard Tutorial has a small example on how to visualize the model. :slight_smile: