mxnet学習二、基礎(NDArrayを理解)

4091 ワード

文書ディレクトリ
  • 一、学習シリーズ転送ドア
  • 二、NDArray
  • を理解する
    このブログは『MXNet深さ学習実戦』のシリーズノートを参考にしています
    一、学習シリーズ転送ゲート
    前のメモ:1、mxnet学習一、データ作成及び読み取り
    二、NDArrayを知る
    NDArrayはmxnetフレームワークにおけるデータストリームのインフラストラクチャであり,python下のnumpyと非常に類似しており,GPU下のnumpyと見なすことができる.1、NDArrayを定義する
    a = mx.nd.array([[1,2],[4,3]])
    

    しゅつりょく
    [[1. 2.]
     [4. 3.]]
    (0)>
    

    2、NDArrayとnumpyの変換
    NDArrayはnumpyを回し、NDArrayのasnumpy()オブジェクトを呼び出す
    a = mx.nd.array([[1,2],[3,4]])
    b = a.asnumpy()
    

    numpy转NDArray,利用mx.nd.array()
    a = np.array([[1,2],[3,4]])
    b = mx.nd.array(a)
    

    3、どのカードで実行されているNDArrayを確認する
    print(b.context)
    

    4、NDArrayオブジェクトを利用したas_in_context()
    a = a.as_in_context(mx.gpu(0))