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

来源:2-2 【讨论题】比较动态图与静态图推理框架

谢思阳

2021-09-30

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

写回答

1回答

会写代码的好厨师

2021-10-11

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

0
1
谢思阳
老师您好,我明白叶子节点的梯度会被保存而非叶子节点不会。我的问题是`torch.autograd.grad`是否会自动判断节点是否是叶子节点,如果是,就计算grad;如果不是,就计算grad_fn?
2021-10-11
共1条回复

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程