Integrating Tensorboard into My Neural Network

I’m working on a neural network that uses Fourier transform in order to handle complex numbers. I want to implement Tensorboard for this network so that I can better see the graph of this network and how the data goes from one layer to the next. However, I’ve been struggling to figure out how exactly to implement it. I’ve read through a lot of Tensorboard tutorials, but I feel that it’s difficult to translate the simple examples a lot of them have to the code I have.

Here is my code:

If anyone has any advice, please let me know.

1 Like