[ 仅 API 调用方式不一致 ]torch.nn.Module.modules

torch.nn.Module.modules

torch.nn.Module.modules()

paddle.nn.Layer.sublayers

paddle.nn.Layer.sublayers(include_self=False)

两者功能一致,但调用方式不一致,具体如下:

转写示例

# PyTorch 写法
model = torch.nn.Sequential(net1, net2)
for layer in model.modules():
    print(layer)

# Paddle 写法
model = paddle.nn.Sequential(net1, net2)
for layer in model.sublayers():
    print(layer)