数据操作:
对元素进行加减乘除
矩阵X = 矩阵Y=
X+Y
X*Y 是元素乘法
X / Y
如果是矩阵乘法的话就是nd.dot(X,Y.T) dot是点乘函数,‘.T’是转置
自动求导:
比如对 y=2*x*x 进行求导
(1)首先创建x,赋予初值
(2)申请空间存x的导数
(3)用record()函数来显示要求MXNET记录求导的程序
(4)反向传播
(5)求得导数
from mxnet import ndarray as nd
from mxnet import autograd as ag
x=nd.array([[1,2],[3,4]]) #第(1)步
x.attach_grad() #第(2)步
with ag.record(): #第(3)步
y = 2 * x
z = y * x
z.backward() #第(4)步
x.grad #第(5)步
x.grad == 4 * x #判断求导是否正确
对控制流求导
norm() 对所有元素平方和开方