Параметры для настройки генерации кода глубокого обучения с помощью библиотеки CUDA Глубокой Нейронной Сети
The coder.CuDNNConfig
объект содержит NVIDIA® cuDNN конкретные параметры, которые codegen
используется для генерации CUDA® код для глубоких нейронных сетей.
Как использовать coder.CuDNNConfig
объект для генерации кода, присвойте его DeepLearningConfig
свойство coder.gpuConfig
объект, которому вы передаете codegen
.
Создайте объект строения cuDNN с помощью coder.DeepLearningConfig
функция с целевым набором библиотек в виде 'cudnn'
.