3-32 是否可以理解为grad函数可以计算grad和grad_fn

Question

是否可以理解为torch.autograd.grad函数可以根据节点是否是叶子节点，自动计算grad和grad_fn？

Answer 1

会写代码的好厨师 2021-10-11 11:10:38

叶子节点和非叶子节点的区别在于，梯度是否会被保存。只保存叶子节点梯度

• 提问者 谢思阳 #1

  老师您好，我明白叶子节点的梯度会被保存而非叶子节点不会。我的问题是`torch.autograd.grad`是否会自动判断节点是否是叶子节点，如果是，就计算grad；如果不是，就计算grad_fn？