Pytorch如何打印每层的输出形状

from torchsummary import summary
from torchvision import models
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
net = models.resnet18(pretrained=True).to(device)
#以上两步必须要有，且顺序不能调换。
summary(net,(3,224,224)) #自动打印。(3,224,224)是输入图像尺寸，3通道，224H，224W

效果