I just… type them out manually…so that you can read more easily…
But if you are interesting in how to print out the output shape of each layer automatically, here’s a simple tutorial about forward hooks 12.