【发布时间】:2021-06-18 04:29:43
【问题描述】:
cuDNN 中是否有与 Pytorch 中的 AdaptiveAvgPool2d 等效的 API?
【问题讨论】:
cuDNN 中是否有与 Pytorch 中的 AdaptiveAvgPool2d 等效的 API?
【问题讨论】:
是的,这是可能的。您可以创建池描述符。
这里是 API 的官方文档- https://docs.nvidia.com/deeplearning/cudnn/api/index.html#cudnnPoolingMode_t
【讨论】: