HarisIqbal88 / PlotNeuralNet

Latex code for making neural networks diagrams
MIT License
21.93k stars 2.86k forks source link

Residual block example #102

Open rodrigovimieiro opened 3 years ago

rodrigovimieiro commented 3 years ago

Hello, thanks for providing all the codes. They are really helpful.

I made a simple residual block example. In order to make it work, I had to slithy modify the to_skip function so it could be added an offset in the summation as well. I also added a ReLU block in the same manner as Pool. See the figure below.

Thanks!

Screen Shot 2021-01-29 at 13 45 13