鲲鹏社区首页
中文
注册
开发者
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

算子说明

KDNN_EXT是鲲鹏深度学习性能加速库KDNN的扩展模块,有以下特点:

  • 接口易用:使用Cython框架,向用户提供Python语言接口,更容易嵌入用户场景。
  • 高性能:底层使用C代码实现,接口性能好。

当前支持以下算子:

  • random_choice算子
  • softmax算子