[ torch 参数更多 ]torch.nn.Module.register_forward_hook¶
torch.nn.Module.register_forward_hook¶
torch.nn.Module.register_forward_hook(hook, *, prepend=False, with_kwargs=False)
paddle.nn.Layer.register_forward_post_hook¶
paddle.nn.Layer.register_forward_post_hook(hook)
PyTorch 参数更多,具体如下:
参数映射¶
| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | hook | hook | 被注册的 hook。 | | prepend | - | 是否在其他 hook 执行前执行,Paddle 无此参数, 暂无转写方式。 | | with_kwargs | - | 是否传入 forward 函数的参数,paddle 无此参数, 暂无转写方式。 |