NDArray的簡單操作

NDArray是MXNet中的一種數據操作工具,類似於Numpy.

創建:

x =nd.arange(6)

[0. 1. 2. 3. 4. 5.]
x = nd.zeros((2,3))

[[0 0 0]

[0 0 0 ]]
根據列表生成NDArray

list1=[1,2,3,4]

x=nd.array(list1)
隨機生成NDArray

nd.random.normal(0,1,shape=(3,4))

隨機生成3*4矩陣,均值=0,方差=1

運算:

在兩個數據shape相同時

支持按元素的+ - * /和exp()

支持 == , >=, <=按元素比較,返回相同shape的0 1數據

矩陣乘法nd.dot(X,Y) X的列和Y的行相同,保證兩個矩陣可以進行乘法計算

shape不同的兩個數據可以使用廣播機制,但是要保證維度相同,如果需要擴充維度可以使用

nd.expand(a,axis=0) 在第0維上擴充維度

Symbol同樣支持廣播機制

支持索引(Symbol不支持索引)

避免臨時內存開銷

 nd.elemwise_add(X, Y, out=Z) 相當於Z=X+Y
或是X+=Y
 

NDArrayNumPy相互變換

numpy轉NDArray:
P = np.ones((2, 3))
D = nd.array(P)
 
NDArray轉Numpy
D.asnumpy()
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章