кодер. MklDNNConfig

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

Описание

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

Чтобы использовать объект coder.MklDNNConfig для генерации кода, присвойте его свойству DeepLearningConfig объекта coder.CodeConfig или coder.EmbeddedCodeConfig, который вы передаете 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

Была ли эта тема полезной?