[ 参数完全一致 ]torch.Tensor.register_hook

torch.Tensor.register_hook

torch.Tensor.register_hook(hook)

paddle.Tensor.register_hook

paddle.Tensor.register_hook(hook)

两者功能一致,且参数一致,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | hook | hook | 一个需要注册到 Tensor.grad 上的 hook 函数。 |