PyTorch中autograd:
链接:autograd.
核心:chain rule–链式规则,用于反向自动梯度微分,y=f(x), l=g(y),如果y是non-scalar的张量,y对x的full Jacobian梯度无法直接计算。但如果只需要计算vector-Jacobian product,可以传递一个vector(该vector是l对y的梯度)到y.backward(vector)中,再使用x.grad获取vector-Jacobian product。
PyTorch中autograd
最新推荐文章于 2023-03-08 14:51:49 发布