개발챙 AI study
/
Pytorch
/
Torch.Tensor.register_hook(hook)
Search
Share
Torch.Tensor.register_hook(hook)
gradient를 구할 때 중간까지만 구하고 싶을 때 사용