NDArray–資料操作功能

NDArray是儲存和變換資料的主要工具。其還提供了GPU計算和自動求梯度等等功能。
以下為寫到test01.py檔案中的程式碼。

from mxnet import nd
x = nd.arange(12)
print(x)
print(x.shape) #獲取NDArray例項形狀
print(x.size) #得到例項中的元素個數

x = x.reshape((3, 4)) #將向量x改變成一個矩陣，3x4的矩陣
print(x)

print(x.sum()) #輸出矩陣x的和，所有元素相加

print(nd.zeros((2, 3, 4))) #建立2x3x4的矩陣

print(nd.ones((3, 4))) #建立單位矩陣

y = nd.array([[2, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]]) #建立指定型別的矩陣
print(y)

x1 = nd.random.normal(0 ,1, shape=(3, 4))
print(x1) #建立形狀為3x4的NDArray,其中的每個元素都是隨機取樣於均值為0，標準差為1的>>
正太分佈

print(x + y) #輸出x與y相加的結果

print(x * y) #輸出x與y相乘的結果

print(x / y) #輸出x與y相除的結果
z = nd.dot(x, y.T) #使用dot函式做矩陣運算，將x與y的轉置做乘法運算

xy01 = nd.concat(x, y, dim = 0) #連線到行上
print(xy01)

xy02 = nd.concat(x, y, dim = 1) #連線到列上
print(xy02)

print(x[1:3]) #行索引，會輸出矩陣x的索引為1與2的兩行

x[1, 2] = 9 #給矩陣的索引為1,2位置重新賦值
print(x)
 

執行程式：python test01.py
得到結果如下：

[ 0.  1.  2.  3.  4.  5.  6.  7.  8.  9. 10. 11.]
<NDArray 12 @cpu(0)>
(12,)
12

[[ 0.  1.  2.  3.]
 [ 4.  5.  6.  7.]
 [ 8.  9. 10. 11.]]
<NDArray 3x4 @cpu(0)>

[[[0. 0. 0. 0.]
  [0. 0. 0. 0.]
  [0. 0. 0. 0.]]

 [[0. 0. 0. 0.]
  [0. 0. 0. 0.]
  [0. 0. 0. 0.]]]
<NDArray 2x3x4 @cpu(0)>

[[1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]]
<NDArray 3x4 @cpu(0)>

[[2. 1. 4. 3.]
 [1. 2. 3. 4.]
 [4. 3. 2. 1.]]
<NDArray 3x4 @cpu(0)>

[[ 2.2122064   0.7740038   1.0434403   1.1839255 ]
 [ 1.8917114  -1.2347414  -1.771029   -0.45138445]
 [ 0.57938355 -1.856082   -1.9768796  -0.20801921]]
<NDArray 3x4 @cpu(0)>

[[ 2.  2.  6.  6.]
 [ 5.  7.  9. 11.]
 [12. 12. 12. 12.]]
<NDArray 3x4 @cpu(0)>

[[ 0.  1.  8.  9.]
 [ 4. 10. 18. 28.]
 [32. 27. 20. 11.]]
<NDArray 3x4 @cpu(0)>

[[ 0.    1.    0.5   1.  ]
 [ 4.    2.5   2.    1.75]
 [ 2.    3.    5.   11.  ]]

 

總結

這些語句都是沒有必要死記的，後續用的多了就能記住。用到那個的時候百度一下即可。熟能生巧