Search

Torch.Tensor.register_hook(hook)

gradient를 구할 때 중간까지만 구하고 싶을 때 사용