[ 参数完全一致 ]torch.Tensor.register_hook¶
torch.Tensor.register_hook¶
torch.Tensor.register_hook(hook)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
hook | hook | 一个需要注册到 Tensor.grad 上的 hook 函数。 |
torch.Tensor.register_hook(hook)
PyTorch | PaddlePaddle | 备注 |
---|---|---|
hook | hook | 一个需要注册到 Tensor.grad 上的 hook 函数。 |