HarisIqbal88 / PlotNeuralNet

Latex code for making neural networks diagrams
MIT License
21.54k stars 2.83k forks source link

Add caption to blocks #123

Open ema2159 opened 2 years ago

ema2159 commented 2 years ago

Hi. While using your package I noticed that it was not possible to add a caption to the currently available blocks. In order to solve this, I modified the code to give the user the option to add a caption to the block, which is added in the layer that's in the middle of such block. Also, I used Black to format the file so it is more readable.

P.D. The first commit is just formatting. You can see the actual functional changes in the second commit.