coder.MklDNNConfig

Параметры, чтобы сконфигурировать генерацию кода глубокого обучения с Math Kernel Library Intel для Глубоких нейронных сетей

Описание

coder.MklDNNConfig объект содержит Intel® MKL-DNN определенные параметры это codegen использование для генерации Кода С++ для глубоких нейронных сетей.

Использовать coder.MklDNNConfig объект для генерации кода, присвойте его DeepLearningConfig свойство настройки генерации кода возражает, что вы передаете codegen.

Создание

Создайте объект настройки MKL-DNN при помощи coder.DeepLearningConfig функция с целевой библиотекой установила как 'mkldnn'.

Свойства

развернуть все

Имя целевой библиотеки в виде вектора символов.

Примеры

свернуть все

Создайте функцию точки входа resnet_predict это использует coder.loadDeepLearningNetwork функционируйте, чтобы загрузить resnet50 (Deep Learning Toolbox) SeriesNetwork объект.

function out = resnet_predict(in)

persistent mynet;
if isempty(mynet)
    mynet = coder.loadDeepLearningNetwork('resnet50', 'myresnet');
end

out = predict(mynet,in);

Создайте coder.config объект настройки для генерации кода MEX.

cfg = coder.config('mex');

Установите выходной язык на C++.

cfg.TargetLang = 'C++';

Создайте coder.MklDNNConfig объект настройки глубокого обучения. Присвойте его DeepLearningConfig свойство cfg объект настройки.

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

Используйте -config опция codegen функционируйте, чтобы передать cfg объект настройки. codegen функция должна определить размер, класс и сложность MATLAB® входные параметры функции. Используйте -args опция, чтобы задать размер входа к функции точки входа.

codegen -args {ones(224,224,3,'single')} -config cfg resnet_predict

codegen команда помещает все сгенерированные файлы в codegen папка. Папка содержит Код С++ для функции точки входа resnet_predict.cpp, заголовок и исходные файлы, содержащие определения класса C++ для сверточной нейронной сети (CNN), веса и файлов смещения.

Введенный в R2018b