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