It is a cool visualization tool for Pytorch. I just wonder whether this support network architecture visualization.