[torch 参数更多]torch.nn.Module.named_modules¶
torch.nn.Module.named_modules¶
torch.nn.Module.named_modules(memo=None, prefix='', remove_duplicate=True)
paddle.nn.Layer.named_sublayers¶
paddle.nn.Layer.named_sublayers(prefix='', include_self=False, layers_set=None)
PyTorch 相比 Paddle 支持更多其他参数，具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| memo | - | 是否用备忘录对结果进行存储。Paddle 无此参数，暂无转写方式。 |
| prefix | prefix | 在所有参数名称前加的前缀。 |
| remove_duplicate | - | 是否删除结果中重复的模块实例, Paddle 无此参数，暂无转写方式。 |
| - | include_self | 是否包含该层自身，PyTorch 无此参数，Paddle 保持默认即可。 |
| - | layers_set | 记录重复子层的集合，PyTorch 无此参数，Paddle 保持默认即可。 |