在CAM中有遇到 register_hook
import torch v = torch.ones((1, 3), dtype=torch.float32, requires_grad=True) y = v**2 z = y.sum() y.register_hook(lambda grad: print(grad)) z.backward() print(v.grad)register_hook是用来对tensor的grad进行操作的
在CAM中有遇到 register_hook
import torch v = torch.ones((1, 3), dtype=torch.float32, requires_grad=True) y = v**2 z = y.sum() y.register_hook(lambda grad: print(grad)) z.backward() print(v.grad)register_hook是用来对tensor的grad进行操作的
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
