这是我的代码。
进口火炬v = torch.tensor([0。,0.,0。],requires_grad = True)x = 1def f(grad): 全球x x = 2 返回毕业* 2h = v.register_hook(f)#使梯度加倍v.backward(…
你可以试试 ipdb https://pypi.org/project/ipdb/ 代替 pdb 。
ipdb
pdb