MXNet快速入門之NDArray詳細介紹

前言

由於最近需要用到mxnet,所以準備結合mxnet官網教程,寫一系列mxnet的快速入門教程。通過一系列文章你將能夠快速掌握mxnet的核心NDArray和Gluron以及靈活利用mxnet來構建神經網絡,mxnet中的NDArray數據結構相當於TensorFlow中的tensor,其結構有點類似於numpy的array,通過mxnet的Gluron可以快速來構建神經網絡。接下來的這篇文章,將詳細介紹一下NDArray。

MXNet是亞馬遜(Amazon)公司的一個深度學習框架,為多GPU配置提供了良好的設置,多個GPU之間的切換隻需要一句代碼。同時使用MXNet來構建一個神經網絡架構也是非常簡單,在後面的教程中會詳細介紹。

mxnet官網教程:https://beta.mxnet.io/guide/crash-course/index.html

MXNet快速入門之NDArray詳細介紹

MXNet安裝教程

mxnet官網安裝教程:http://mxnet.incubator.apache.org/versions/master/install/index.html

根據自己的需要安裝GPU或CPU版本,具體安裝請參考MXNet安裝教程根據自己的系統以及需要安裝的版本等信息進行選擇,然後根據生成的命令安裝,該系列教程主要包括以下內容:

  • MXNet中的核心數據類型NDArray介紹
  • 如何使用MXNet來構建一個神經網絡
  • 利用autograd來計算鏈式求導
  • 訓練神經網絡
  • 利用預訓練模型來進行預測
  • 如何使用GPU加速運算

NDArray

  • 通過元組創建NDArray
MXNet快速入門之NDArray詳細介紹

  • 利用內置函數創建矩陣
MXNet快速入門之NDArray詳細介紹

  • 查看NDArray的相關信息
MXNet快速入門之NDArray詳細介紹

  • NDArray的乘法
MXNet快速入門之NDArray詳細介紹

注意:使用*的乘法,即一對一的乘法,不是矩陣的乘法。所以x的shape需要與z的shape相同或者z的shape為1,當z的shape為1時,在計算乘法的時候z的shape會自動擴展與x相同這種操作被稱為

廣播

  • 將矩陣中的元素求e(自然底數)的次方
MXNet快速入門之NDArray詳細介紹

  • 矩陣的乘法
MXNet快速入門之NDArray詳細介紹

注意:在計算矩陣的乘法時,前一個矩陣的列數必須要等於後一個矩陣的行數。因為要滿足矩陣點乘公式,矩陣之間的乘法計算如下:

MXNet快速入門之NDArray詳細介紹

  • 矩陣的轉置
MXNet快速入門之NDArray詳細介紹

  • NDArray的切片
MXNet快速入門之NDArray詳細介紹

  • 將NDArray轉為numpy的array
MXNet快速入門之NDArray詳細介紹

最近開通了微信公眾號,在微信公眾號上,分享了一些資料,搜索"Python機器學習之路"關注我即可獲取,歡迎來找我玩。


分享到:


相關文章: