钩子函数
来源:3-32 Pytorch与autograd中的几个重要概念-autograd例子

Amberzy
2020-07-29
x = register_hook(lambda grad:grad*2)
这个钩子函数是在计算梯度的时候作用的吗?
写回答
1回答
-
这个操作主要是执行lamda这个函数,也即是梯度*2的时候,会被执行到。
012020-08-21
相似问题
钩子函数
来源:3-32 Pytorch与autograd中的几个重要概念-autograd例子
Amberzy
2020-07-29
x = register_hook(lambda grad:grad*2)
这个钩子函数是在计算梯度的时候作用的吗?
1回答
这个操作主要是执行lamda这个函数,也即是梯度*2的时候,会被执行到。
相似问题