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