torchsummary使用

先导包，pip install torchsummary 然后 ...