PyTorch学习笔记(3)autograd 自动求导
最新推荐文章于 2021-12-21 22:29:00 发布
本文是PyTorch学习笔记的第三部分,主要探讨autograd模块的自动求导功能。通过torch.autograd.backward可以自动计算梯度,支持设置retain_graph以保留计算图,create_graph用于构建高阶导数图,同时可以指定grad_tensors进行多梯度权重计算。需要注意的是,autograd系统中,梯度不会自动清零,并且依赖于requires_grad=True的叶子结点来跟踪计算历史。
摘要由CSDN通过智能技术生成