#pytorch #cudnn
Вопрос:
Существует ли эквивалентный API в cuDNN, как AdaptiveAvgPool2d в Pytorch?
Ответ №1:
да, это возможно. вы можете создать дескриптор объединения.
вот официальная документация по API — https://docs.nvidia.com/deeplearning/cudnn/api/index.html#cudnnPoolingMode_t