지원되는 신경망, 계층 및 클래스

GPU Coder™는 시리즈 신경망 및 유방향 비순환 그래프(DAG) 신경망을 위한 코드 생성을 지원합니다. 이 페이지에서는 지원되는 딥러닝 신경망, 계층 및 클래스 목록을 제공합니다.

참고로, 코드 생성기는 문자를 로캘 설정에 따라 결정되는 8비트 ASCII 코드 집합으로 표현합니다. 따라서 클래스 이름, 계층 이름, 계층 설명 또는 신경망 이름에 비ASCII 문자를 사용하면 오류가 발생할 수 있습니다. 자세한 내용은 코드 생성 시 문자 인코딩 항목을 참조하십시오.

지원되는 사전 훈련된 신경망

Deep Learning Toolbox™를 사용하여 하나의 CPU, 하나의 GPU 또는 복수의 GPU에서 신경망을 훈련시키거나 표에 나열된 사전 훈련된 신경망을 사용하여 CUDA^® 코드를 생성할 수 있습니다.

아래의 사전 훈련된 신경망은 명시된 타깃 딥러닝 라이브러리에 대해 GPU Coder의 코드 생성이 지원됩니다.

- 지원에 제한이 있을 수 있습니다. 오류 및 예기치 않은 동작이 발생할 수 있습니다. 자세한 내용은 함수 도움말 페이지의 확장 기능 섹션을 참조하십시오.

신경망 이름	CUDA	TensorRT	cuDNN
`alexnet` (Deep Learning Toolbox)
Caffe 신경망 `importCaffeNetwork` (Deep Learning Toolbox) 항목을 참조하십시오.
`darknet19` (Deep Learning Toolbox)
`darknet53` (Deep Learning Toolbox)
`deeplabv3plusLayers` (Computer Vision Toolbox)
`densenet201` (Deep Learning Toolbox)
`efficientnetb0` (Deep Learning Toolbox)
`googlenet` (Deep Learning Toolbox)
`inceptionresnetv2` (Deep Learning Toolbox)
`inceptionv3` (Deep Learning Toolbox)
`mobilenetv2` (Deep Learning Toolbox)
`nasnetlarge` (Deep Learning Toolbox)
`nasnetmobile` (Deep Learning Toolbox)
`resnet18` (Deep Learning Toolbox)
`resnet50` (Deep Learning Toolbox)
`resnet101` (Deep Learning Toolbox)
`segnetLayers` (Computer Vision Toolbox)
`squeezenet` (Deep Learning Toolbox)
`vgg16` (Deep Learning Toolbox)
`vgg19` (Deep Learning Toolbox)
`xception` (Deep Learning Toolbox)
`yolov2Layers` (Computer Vision Toolbox)

지원되는 계층

다음 계층은 명시된 타깃 딥러닝 라이브러리에 대해 GPU Coder의 코드 생성이 지원됩니다.

입력 계층

계층 이름	CUDA	TensorRT	cuDNN
`imageInputLayer` (Deep Learning Toolbox)
`sequenceInputLayer` (Deep Learning Toolbox)
`featureInputLayer` (Deep Learning Toolbox)

컨벌루션 계층과 완전 연결 계층

계층 이름	CUDA	TensorRT	cuDNN
`convolution1dLayer` (Deep Learning Toolbox)
`convolution2dLayer` (Deep Learning Toolbox)
`fullyConnectedLayer` (Deep Learning Toolbox)
`groupedConvolution2dLayer` (Deep Learning Toolbox)
`transposedConv2dLayer` (Deep Learning Toolbox)

시퀀스 계층

계층 이름	CUDA	TensorRT	cuDNN
`sequenceInputLayer` (Deep Learning Toolbox)
`bilstmLayer` (Deep Learning Toolbox)
`flattenLayer` (Deep Learning Toolbox)
`gruLayer` (Deep Learning Toolbox)
`lstmLayer` (Deep Learning Toolbox)
`sequenceFoldingLayer` (Deep Learning Toolbox)
`sequenceUnfoldingLayer` (Deep Learning Toolbox)
`wordEmbeddingLayer` (Text Analytics Toolbox)

활성화 계층

계층 이름	CUDA	TensorRT	cuDNN
`clippedReluLayer` (Deep Learning Toolbox)
`eluLayer` (Deep Learning Toolbox)
`leakyReluLayer` (Deep Learning Toolbox)
`reluLayer` (Deep Learning Toolbox)
`gelu` (Deep Learning Toolbox)
`softplusLayer` (Reinforcement Learning Toolbox)
`swishLayer` (Deep Learning Toolbox)
`tanhLayer` (Deep Learning Toolbox)

정규화, 드롭아웃 및 자르기 계층

계층 이름	CUDA	TensorRT	cuDNN
`batchNormalizationLayer` (Deep Learning Toolbox)
`crossChannelNormalizationLayer` (Deep Learning Toolbox)
`groupNormalizationLayer` (Deep Learning Toolbox)
`layerNormalizationLayer` (Deep Learning Toolbox)
`crop2dLayer` (Deep Learning Toolbox)
`dropoutLayer` (Deep Learning Toolbox)
`scalingLayer` (Reinforcement Learning Toolbox)

풀링 및 언풀링 계층

계층 이름	CUDA	TensorRT	cuDNN
`averagePooling1dLayer` (Deep Learning Toolbox)
`averagePooling2dLayer` (Deep Learning Toolbox)
`globalAveragePooling1dLayer` (Deep Learning Toolbox)
`globalAveragePooling2dLayer` (Deep Learning Toolbox)
`globalMaxPooling1dLayer` (Deep Learning Toolbox)
`globalMaxPooling2dLayer` (Deep Learning Toolbox)
`maxPooling1dLayer` (Deep Learning Toolbox)
`maxPooling2dLayer` (Deep Learning Toolbox)
`maxUnpooling2dLayer` (Deep Learning Toolbox)

결합 계층

계층 이름	CUDA	TensorRT	cuDNN
`additionLayer` (Deep Learning Toolbox)
`concatenationLayer` (Deep Learning Toolbox)
`depthConcatenationLayer` (Deep Learning Toolbox)

트랜스포머 계층

계층 이름	CUDA	TensorRT	cuDNN
`AttentionLayer` (Deep Learning Toolbox)
`embeddingConcatenationLayer` (Deep Learning Toolbox)
`indexing1dLayer` (Deep Learning Toolbox)
`patchEmbeddingLayer` (Computer Vision Toolbox)
`PositionEmbeddingLayer` (Deep Learning Toolbox)
`selfAttentionLayer` (Deep Learning Toolbox)

객체 검출 계층

계층 이름	CUDA	TensorRT	cuDNN
`anchorBoxLayer` (Computer Vision Toolbox)
`depthToSpace2dLayer` (Image Processing Toolbox)
`spaceToDepthLayer` (Image Processing Toolbox)
`ssdMergeLayer` (Computer Vision Toolbox)
`yolov2TransformLayer` (Computer Vision Toolbox)
`rcnnBoxRegressionLayer` (Computer Vision Toolbox)
`focalLossLayer` (Computer Vision Toolbox)
`rpnClassificationLayer` (Computer Vision Toolbox)
`yolov2OutputLayer` (Computer Vision Toolbox)

출력 계층

계층 이름	CUDA	TensorRT	cuDNN
`classificationLayer` (Deep Learning Toolbox)
`dicePixelClassificationLayer` (Computer Vision Toolbox)
`focalLossLayer` (Computer Vision Toolbox)
`pixelClassificationLayer` (Computer Vision Toolbox)
`rcnnBoxRegressionLayer` (Computer Vision Toolbox)
`regressionLayer` (Deep Learning Toolbox)
`rpnClassificationLayer` (Computer Vision Toolbox)
`sigmoidLayer` (Deep Learning Toolbox)
`softmaxLayer` (Deep Learning Toolbox)
`Custom Output Layer` (Deep Learning Toolbox) 추가 정보 `nnet.layer.ClassificationLayer` 함수로 생성된 사용자 지정 분류 출력 계층 또는 `nnet.layer.RegressionLayer` 함수로 생성된 사용자 지정 회귀 출력 계층을 포함한 출력 계층. 사용자 지정 분류 출력 계층을 정의하고 손실 함수를 지정하는 방법을 보여주는 예제는 사용자 지정 분류 출력 계층 정의하기 (Deep Learning Toolbox) 항목을 참조하십시오. 사용자 지정 회귀 출력 계층을 정의하고 손실 함수를 지정하는 방법을 보여주는 예제는 사용자 지정 회귀 출력 계층 정의하기 (Deep Learning Toolbox) 항목을 참조하십시오.

사용자 지정 Keras 계층

계층 이름	CUDA	TensorRT	cuDNN
`nnet.keras.layer.ClipLayer` (Deep Learning Toolbox)
`nnet.keras.layer.FlattenCStyleLayer` (Deep Learning Toolbox)
`nnet.keras.layer.GlobalAveragePooling2dLayer` (Deep Learning Toolbox)
`nnet.keras.layer.PreluLayer` (Deep Learning Toolbox)
`nnet.keras.layer.SigmoidLayer` (Deep Learning Toolbox)
`nnet.keras.layer.TanhLayer` (Deep Learning Toolbox)
`nnet.keras.layer.TimeDistributedFlattenCStyleLayer` (Deep Learning Toolbox)
`nnet.keras.layer.ZeroPadding2dLayer` (Deep Learning Toolbox)

사용자 지정 ONNX 계층

계층 이름	CUDA	TensorRT	cuDNN
`nnet.onnx.layer.ClipLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.ElementwiseAffineLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.FlattenInto2dLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.FlattenLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.GlobalAveragePooling2dLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.IdentityLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.PreluLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.SigmoidLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.TanhLayer` (Deep Learning Toolbox)
`nnet.onnx.layer.VerifyBatchSizeLayer` (Deep Learning Toolbox)

사용자 지정 계층

계층 이름 CUDA TensorRT cuDNN

계층 이름	CUDA	TensorRT	cuDNN
`Custom layers` 추가 정보 사용자가 문제에 대해 정의하는 사용자 지정 계층(학습 가능한 파라미터를 포함할 수도 있고 포함하지 않을 수도 있음). 사용자 지정 딥러닝 계층을 정의하는 방법은 사용자 지정 딥러닝 계층 정의하기 (Deep Learning Toolbox) 및 Define Custom Deep Learning Layer for Code Generation (Deep Learning Toolbox) 항목을 참조하십시오. 사용자 지정 계층이 있는 신경망에 대한 코드를 생성하는 방법에 대한 예제는 YOLO v3 딥러닝 신경망을 사용한 객체 검출 코드 생성 항목을 참조하십시오. 코드 생성 제한 사항 사용자 지정 계층의 출력은 고정 크기 배열이어야 합니다. `coder.gpuConfig`에서 `MallocMode` 속성을 `"unified"`로 설정하면 추가 메모리 복사가 필요하므로 성능이 더 느려집니다. `"discrete"`를 사용할 것을 권장합니다. GPU 메모리 할당에 대한 자세한 내용은 Discrete and Managed Modes 항목을 참조하십시오. cuDNN 타깃은 사용자 지정 계층에 대해 행 우선 방식과 열 우선 방식의 코드 생성을 둘 다 지원합니다. TensorRT 타깃은 열 우선 코드 생성만 지원합니다. 사용자 지정 계층에는 `%#codegen` 프라그마가 포함되어야 합니다. 사용자 지정 계층과 LSTM 또는 GRU 계층을 포함하는 시퀀스 신경망은 지원되지 않습니다. 다음과 같은 경우 사용자 지정 계층에 `dlarray`를 전달할 수 있습니다. 사용자 지정 계층이 `dlnetwork`에 있습니다. 사용자 지정 계층이 DAG 또는 시리즈 신경망에 있으며 `nnet.layer.Formattable`에서 상속되거나 역방향 전파가 없습니다. 지원되지 않는 `dlarray` 메서드의 경우 `dlarray`에서 기본 데이터를 추출하고 계산을 수행한 후 데이터를 `dlarray` 객체로 재구성해야 합니다. 예를 들면 다음과 같습니다. function Z = predict(layer, X) if coder.target('MATLAB') Z = doPredict(X); else if isdlarray(X) X1 = extractdata(X); Z1 = doPredict(X1); Z = dlarray(Z1); else Z = doPredict(X); end end end

Custom layers

추가 정보

코드 생성 제한 사항

지원되는 클래스

다음 클래스는 명시된 타깃 딥러닝 라이브러리에 대해 GPU Coder의 코드 생성이 지원됩니다.

이름	CUDA	TensorRT	cuDNN
`dlnetwork` (Deep Learning Toolbox)
`DAGNetwork` (Deep Learning Toolbox)
`pointPillarsObjectDetector` (Lidar Toolbox)
`SeriesNetwork` (Deep Learning Toolbox)
`ssdObjectDetector` (Computer Vision Toolbox)
`yolov2ObjectDetector` (Computer Vision Toolbox)
`yolov3ObjectDetector` (Computer Vision Toolbox)
`yolov4ObjectDetector` (Computer Vision Toolbox)
`yoloxObjectDetector` (Computer Vision Toolbox)