Exponenta Banner

coder.getDeepLearningLayers

Поддержите слои сверточной нейронной сети для генерации кода для определенной библиотеки глубокого обучения

свернуть все на странице

Синтаксис

coder.getDeepLearningLayers(libraryname)

Описание

пример

coder.getDeepLearningLayers(libraryname) возвращает слои сверточной нейронной сети, поддержанные для генерации кода для определенной библиотеки глубокого обучения.

Примечание

Использовать coder.getDeepLearningLayers, необходимо установить пакет поддержки, который соответствует libraryname:

  • Для 'arm-compute' и 'mkldnn', установите MATLAB® Coder™ Interface для Библиотек Глубокого обучения.

  • Для 'cudnn' и 'tensorrt', установите Интерфейс GPU Coder™ для Библиотек Глубокого обучения.

Примеры

свернуть все

Получите список слоев, поддержанных для генерации кода для Intel® Math Kernel Library для Глубоких нейронных сетей.

coder.getDeepLearningLayers('mkldnn')
ans =

  17×1 cell array

    {'AdditionLayer'                 }
    {'AveragePooling2DLayer'         }
    {'BatchNormalizationLayer'       }
    {'ClassificationOutputLayer'     }
    {'ClippedReLULayer'              }
    {'Convolution2DLayer'            }
    {'CrossChannelNormalizationLayer'}
    {'DepthConcatenationLayer'       }
    {'DropoutLayer'                  }
    {'FullyConnectedLayer'           }
    {'ImageInputLayer'               }
    {'LeakyReLULayer'                }
    {'MaxPooling2DLayer'             }
    {'ReLULayer'                     }
    {'RegressionOutputLayer'         }
    {'SoftmaxLayer'                  }
    {'TransposedConvolution2DLayer'  }

Входные параметры

свернуть все

Имя библиотеки глубокого обучения, заданной как одно из значений в этой таблице.

ЗначениеОписание
'arm-compute'

Библиотека ARM® Compute

Требует интерфейса MATLAB Coder для библиотек глубокого обучения.

'cudnn'

библиотека cuDNN

Требует продукта GPU Coder и Интерфейса GPU Coder для Библиотек Глубокого обучения.

'mkldnn'

Intel Math Kernel Library для глубоких нейронных сетей

Требует интерфейса MATLAB Coder для библиотек глубокого обучения.

'tensorrt'

Библиотека TensorRT™

Требует продукта GPU Coder и Интерфейса GPU Coder для Библиотек Глубокого обучения.

Смотрите также

Темы

Введенный в R2018b

Документация MATLAB Coder
Поддержка