mxnet2——NDArray and autograd

form mxnet import ndarray as nd

NDArray

0 矩陣

nd.zeros((3,4))

1 矩陣

nd.ones((3,4))

創建矩陣

nd.array([[1,2],[3,4]])

隨機矩陣 均值 a 方差 b nxm

nd.random_normal(a,b,shape=(n,m))

大小 shape

個數 size

運算

+ × 
指數  nd.exp()
點乘  nd.dot()
轉置  y.T

廣播

形狀不一樣，複製

與NumPy的轉換

import numpy as np
x = np.ones((2,3))
y = nd.array(x)  # numpy -> mxnet
z = y.asnumpy()  # mxnet -> numpy

替換

通過[:]寫到一個之前開好的數組裏
z[:] = x + y

截取

從a到b行：x[a:b]
指定位置：x[a,b]
多維：x[1:2,1:3]

autograd

import mxnet.ndrray as nd
import mxnet.autograd as ag

申請空間

x.attah_grad()

顯式求導的程序

with ag.record():
    y = x * 2
    z = y * x

進行求導

z.backward()