How to setup the code for getting a visual representation of the output 3d shape that the network generates.