Draw nice looking architectures of your neural network using this tikz macro.

Example output:

Example output for Z: