mxnet深度学习框架的求导

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_27492735/article/details/82704190

mxnet深度学习框架是一个很优秀的框架，其中包括了很多的模型。

'coding = utf-8'
import mxnet.ndarray as nd
import mxnet.autograd as ag
#对f = 2 * (x ** 2)求关于x的导数，我们先创建变量x，并赋初值
x = nd.array([[1,2],[3,4]])#创建一个矩阵x
x.attach_grad()#创建一个a来储存x的导数
with ag.record():
    y = x * 2
    z = y * x

z.backward()#使用z.backward进行求导
print(x.grad)#打印导数

这里的mxnet可以通过pip安装

pip install mxnet

运行结果如下所示：

"D:/python project/deep learing/numpy/1.2.py"
[[ 4.  8.]
 [12. 16.]]
<NDArray 2x2 @cpu(0)>

Process finished with exit code 0