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

torch.nn.Module.register_buffer

torch.nn.Module.register_buffer(name, tensor, persistent=True)

paddle.nn.Layer.register_buffer

paddle.nn.Layer.register_buffer(name, tensor, persistable=True)

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

参数映射

PyTorch PaddlePaddle 备注
name name 注册 buffer 的名字。可以通过此名字来访问已注册的 buffer。
tensor tensor 将被注册为 buffer 的变量。
persistent persistable 注册的 buffer 是否需要可持久性地保存到 state_dict 中。仅参数名不一致。