pytorch hook学习 register_hook import torch x = torch.Tensor([0,1,2,3]).requires_grad_() y = torch.Tensor([4,5,6,7]).requires_grad_() w = torch.Tensor([1,2,3,4]).requires_grad_() z = x+y; o = w.matmul(z) # o = w(x+y) 中间变量z o.backward() print(x.grad,y.g…