draw_net bug (very minor)

[christopher-beckham]

Hi,

It would be nice to have this as part of the Recipes repo:

https://gist.github.com/ebenolson/1682625dc9823e27d771

BTW, in order for that code to work now, get_output_shape must be replaced with output_shape

Cheers,
Chris