Pytorch之summary 导入summary from torchsummary import summary summary(model , input_size = (channel , H , W))