Custom stages

[Babaee]

Hello,

First of all thank you for building this wonderful package. How can we define a custom stage, like batch normalization, up sampling, etc.

[daitranskku]

you can define by creating more \def and \pic. Read and understand the example code first, then you can modify whatever you want. Good luck