mindspore.ops
模块提供的grad
和value_and_grad
接口可以生成网络模型的梯度。grad
计算网络梯度,value_and_grad
同时计算网络的正向输出和梯度。本次主要练习如何使用grad
接口的主要功能,包括一阶、二阶求导,单独对输入或网络权重求导,返回辅助变量,以及如何停止计算梯度。
昇思25天学习打卡营第15天|高级自动微分
最新推荐文章于 2024-08-21 23:32:17 发布
mindspore.ops
模块提供的grad
和value_and_grad
接口可以生成网络模型的梯度。grad
计算网络梯度,value_and_grad
同时计算网络的正向输出和梯度。本次主要练习如何使用grad
接口的主要功能,包括一阶、二阶求导,单独对输入或网络权重求导,返回辅助变量,以及如何停止计算梯度。