51工具盒子

依楼听风雨
笑看云卷云舒,淡观潮起潮落

微分

PyTorch 自动微分模块

PyTorch 自动微分模块

厉飞雨 阅读(6) 评论(0) 赞(0)

自动微分(Autograd)模块对张量做了进一步的封装,具有自动求导功能。自动微分模块是构成神经网络训练的必要模块,在神经网络的反向传播过程中,Autograd 模块基于正向计算的结果对当前的参数进行微分计算,从而实现网络权重参数的更新。 1. **梯度基本计算** {#title-0} ======================== 我们使用 backward 方法...