Main Content

coder.MklDNNConfig

IntelMath Kernel Library for Deep Neural Networks を使用して深層学習のコード生成を構成するパラメーター

説明

coder.MklDNNConfigオブジェクトには、codegenがディープ ニューラル ネットワークの C++ コードの生成に使用する Intel®MKL-DNN 固有のパラメーターが含まれています。

コード生成にcoder.MklDNNConfigオブジェクトを使用するには,そのオブジェクトを,codegenに渡すコード生成構成オブジェクトのDeepLearningConfigプロパティに割り当てます。

作成

関数coder.DeepLearningConfigでターゲット ライブラリを'mkldnn'と設定して、MKL-DNN 構成オブジェクトを作成します。

プロパティ

すべて展開する

ターゲット ライブラリの名前。文字ベクトルとして指定します。

すべて折りたたむ

関数coder.loadDeepLearningNetworkを使用するエントリポイント関数resnet_predictを作成し、resnet50(Deep Learning Toolbox)SeriesNetworkオブジェクトを読み込みます。

functionout = resnet_predict(in)persistentmynet;ifisempty(mynet) mynet = coder.loadDeepLearningNetwork('resnet50','myresnet');endout = predict(mynet,in);

MEX コード生成用のcoder.config構成オブジェクトを作成します。

cfg = coder.config('mex');

ターゲット言語を C++ に設定します。

cfg.TargetLang ='C++';

coder.MklDNNConfig深層学習構成オブジェクトを作成します。このオブジェクトをcfg構成オブジェクトのDeepLearningConfigプロパティに割り当てます。

cfg.DeepLearningConfig = coder.DeepLearningConfig('mkldnn');

関数codegen-configオプションを使用してcfg構成オブジェクトを渡します。関数codegenは、MATLAB®関数入力のサイズ、クラス、実数/複素数を決定しなければなりません。-argsオプションを使用して、エントリポイント関数への入力のサイズを指定します。

codegen-args{ones(224,224,3,'single')}-configcfgresnet_predict

codegenコマンドは生成されたすべてのファイルをcodegenフォルダーに入れます。このフォルダーには、エントリポイント関数resnet_predict.cppの C++ コード、ヘッダー、畳み込みニューラル ネットワーク (CNN) の C++ クラス定義が含まれるソース ファイル、重み、バイアス ファイルが含まれます。

バージョン履歴

R2018b で導入