coder.getDeepLearningLayers

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

Описание

пример

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

Примечание

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

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

  • Для 'cudnn', 'tensorrt' или 'arm-compute-mali', установите Интерфейс 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 для предназначения для процессоров ARM CPU.

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

'arm-compute-mali'

ARM Вычисляет Библиотеку для предназначения для процессоров ARM GPU.

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

'cudnn'

библиотека cuDNN.

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

'mkldnn'

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

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

'tensorrt'

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

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

Введенный в R2018b