从 python 中的 torch 获取所有层，包括运算符

Question

我想得到所有的层，比如卷积等......以及像 python 中的“add”这样的运算符，但我真的不知道怎么做。

这是我的代码

import torch

# An instance of your model.
model = torch.jit.load('best5.torchscript')

# Print all the layers
for name, param in model.named_parameters():
    print(name, param.shape)

它工作正常，但问题是我只能看到像卷积这样的层，而不是所有的运算符。我该如何解决？

例如在这张图片中，当模型加载到 netron 时，我可以看到所有内容

Answer 1

将

named_parameters

替换为

named_modules

并打印

module

（在您的情况下为

param

）而不是

param.shape