Add more debugging helper to net transformer (#19176)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19176
Add some amenities for debugging.
Reviewed By: llyfacebook
Differential Revision: D14901740
fbshipit-source-id: 2c4018fdbf7e3aba2a754b6b4103a72893c229c2