python
with assert
mxnet NDArray
.norm() .concat() .attach_grad() .backward() .grad()
mxnet autograd
.record() .is_training()
小结
MXNet提供autograd模块来自动化求导过程。
MXNet的autograd模块可以对一般的命令式程序进行求导。
MXNet的运行模式包括训练模式和预测模式。我们可以通过autograd.is_training()来判断运行模式。
练习
在本节对控制流求梯度的例子中,把变量a改成一个随机向量或矩阵。此时计算结果c不再是标量,运行结果将有何变化?该如何分析该结果?
重新设计一个对控制流求梯度的例子。运行并分析结果。