torch中的梯度操作,如何在tensor中实现
收藏
# 在pytorch中
grad是一个梯度打开的tensor
grad = - delta.grad.detach()
d = delta
...
delta.data = d
delta.grad.zero_()
# 我在Paddle中
grad = - delta.grad.detach()
d = delta
...
delta = d
delta.clear_grad()
这样对吗?
0
收藏
请登录后评论