Pytorch如何打印与Keras的model.summary()类似的输出

1 Keras的model.summary()

2 Pytorch实现

2.1 安装torchsummary包

pip install torchsummary

2.2 代码

import torch
import torch.nn as nn
import torch.nn.functional as F
from torchsummary import summary

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__(