[ 仅参数名不一致 ]torch.nn.Module.register_module

torch.nn.Module.register_module

torch.nn.Module.register_module(name, module)

paddle.nn.Layer.add_sublayer

paddle.nn.Layer.add_sublayer(name, sublayer)

两者功能一致,仅参数名不一致,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | name | name | 注册 buffer 的名字。可以通过此名字来访问已注册的 buffer。 | | module | sublayer | 将被注册为 buffer 的模块。仅参数名不一致。 |