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同樣支持廣播機制