主要内容

coder.MklDNNConfig

参数配置与深度学习代码生成英特尔数学核库的深度神经网络

描述

coder.MklDNNConfig对象包含Intel®MKL-DNN的特定参数codegen用于为深度神经网络生成c++代码。

使用一个coder.MklDNNConfig对象进行代码生成时,将其分配给DeepLearningConfig要传递到的代码生成配置对象的属性codegen

创建

描述

例子

deepLearningCfg= coder.DeepLearningConfig (“mkldnn”)创建一个coder.MklDNNConfig对象,用于使用MKL-DNN库生成深度学习代码。

属性

全部展开

目标库的名称,指定为字符向量。

例子

全部折叠

创建一个入口点函数resnet_predict使用coder.loadDeepLearningNetwork函数来加载resnet50(深度学习工具箱)SeriesNetwork对象。

函数= resnet_predict(中)持续的mynet;如果mynet = code . loaddeeplearningnetwork (“resnet50”,“myresnet”);结束=预测(mynet,);

创建一个coder.config用于生成MEX代码的配置对象。

cfg = coder.config (墨西哥人的);

将目标语言设置为c++。

cfg。TargetLang =“c++”;

创建一个coder.MklDNNConfig深度学习配置对象。把它分配给DeepLearningConfig财产的cfg配置对象。

cfg。DeepLearningConfig = coder.DeepLearningConfig (“mkldnn”);

使用配置选择的codegen函数,以传递cfg配置对象。的codegen函数必须确定大小,类,和MATLAB的复杂性®函数的输入。使用arg游戏选项指定入口点函数的输入的大小。

codegenarg游戏{(224224 3,“单”)}配置cfgresnet_predict

codegen命令将所有生成的文件放置到codegen文件夹中。该文件夹包含入口点函数的c++代码resnet_predict.cpp、头文件和源文件,其中包含卷积神经网络(CNN)的c++类定义、权值和偏差文件。

介绍了R2018b