кодер. MklDNNConfig

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

Описание

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

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

Создание

Синтаксис

deepLearningCfg = coder.DeepLearningConfig('mkldnn')

Описание

пример

deepLearningCfg = coder.DeepLearningConfig('mkldnn') создает объект coder.MklDNNConfig для генерации кода глубокого обучения при помощи библиотеки MKL-DNN.

Свойства

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

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

Примеры

свернуть все

Создайте функцию точки входа alexneteg, который использует функцию coder.loadDeepLearningNetwork, чтобы загрузить объект alexnet SeriesNetwork.

function out = alexneteg(in)

persistent mynet;
if isempty(mynet)
    mynet = coder.loadDeepLearningNetwork('alexnet', 'myalexnet');
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(227,227,3,'single')} -config cfg alexneteg

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

Введенный в R2018b